Anomaly Detection Reenabled

manole-alexandru · Apr 10, 2023 · 2975086 · 2975086
1 parent dcd4227
commit 2975086
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train.py b/train.py
@@ -269,6 +269,7 @@ def train(hyp, opt, device, callbacks):  # hyp is path/to/hyp.yaml or hyp dictio
                 f"Logging results to {colorstr('bold', save_dir)}\n"
                 f'Starting training for {epochs} epochs...')
 
+    torch.autograd.set_detect_anomaly(True)
     for epoch in range(start_epoch, epochs):  # epoch ------------------------------------------------------------------
         callbacks.run('on_train_epoch_start')
         model.train()