turn off detect_anomaly

configs/experiment/fm_tops.yaml

@@ -16,13 +16,13 @@ defaults:
 
 tags: ["flow_matching", "tops", "30", "test_runs"]
 
-#seed: 12345
+seed: 12345
 
 trainer:
   min_epochs: 3
   max_epochs: 30
   gradient_clip_val: 0.5
-  detect_anomaly: true # raise exception if NaN or +/-inf is detected in any tensor
+  # detect_anomaly: true # raise exception if NaN or +/-inf is detected in any tensor
 
 model:
   optimizer:
@@ -76,7 +76,7 @@ data:
     - "t"
     #- "w"
     #- "z"
-  batch_size: 256
+  batch_size: 1024
   num_particles: ${model.num_particles}
   variable_jet_sizes: ${model.mask}
   centering: True
@@ -102,7 +102,7 @@ callbacks:
   jetnet_eval:
     every_n_epochs: 3
     num_jet_samples: 2
-    batch_size: 1024
+    batch_size: 256
     plot_w_dists: False
     w_dists_batches: 2
     log_w_dists: True