update number of epochs

jpata · Jul 1, 2021 · dc0dae6 · dc0dae6
1 parent 3016d75
commit dc0dae6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/parameters/cms-gnn-dense-focal.yaml b/parameters/cms-gnn-dense-focal.yaml
@@ -46,7 +46,7 @@ setup:
   batch_size: 5
   num_events_train: 80000
   num_events_test: 10000
-  num_epochs: 20
+  num_epochs: 500
   num_val_files: 100
   dtype: float32
   sample_weights: signal_only