Update base.yaml

mila-iqia · Jan 15, 2025 · 40ff390 · 40ff390
1 parent 7936166
commit 40ff390
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/config/base.yaml b/config/base.yaml
@@ -759,7 +759,6 @@ pna:
 
 dimenet:
   inherits: _geo_gnn
-  max_duration: 360000
   argv:
     --model: 'DimeNet'
     --num-samples: 100000
@@ -921,4 +920,4 @@ cleanrljax:
     --num_envs: auto({cpu_per_gpu}, 128)
     --num_steps: 128
     --num_minibatches: 4
-    --total_timesteps: 10000000
+    --total_timesteps: 10000000