reproducibility update

geometric-intelligence · May 8, 2024 · c136653 · c136653
1 parent 236e8d5
commit c136653
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/configs/train.yaml b/configs/train.yaml
@@ -4,7 +4,7 @@
 # order of defaults determines the order in which configs override each other
 defaults:
   - _self_
-  - dataset: us_country_demos
+  - dataset: ZINC
   - model: graph/gcn #hypergraph/unignn2 #
   - evaluator: default
   - callbacks: default

diff --git a/configs/trainer/default.yaml b/configs/trainer/default.yaml
@@ -3,7 +3,7 @@ _target_: lightning.pytorch.trainer.Trainer
 default_root_dir: ${paths.output_dir}
 
 min_epochs: 1 # prevents early stopping
-max_epochs: 350
+max_epochs: 5
 #accumulate_grad_batches: 1 #${dataset.parameters.batch_size}
 
 accelerator: gpu