Updated example.

mila-iqia · Apr 5, 2024 · 1147d2e · 1147d2e
1 parent 229c009
commit 1147d2e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/examples/local/config_diffusion.yaml b/examples/local/config_diffusion.yaml
@@ -1,14 +1,14 @@
 # general
-loss: cross_entropy
-max_epoch: 5
+max_epoch: 50
+log_every_n_steps: 1
 exp_name: exp_example
 # set to null to avoid setting a seed (can speed up GPU computation, but
 # results will not be reproducible)
 seed: 1234
 
 # data
 data:
-  batch_size: 32
+  batch_size: 1024
   num_workers: 0
   max_atom: 512
 
@@ -31,4 +31,7 @@ optimizer:
 early_stopping:
   metric: val_loss
   mode: min
-  patience: 3
+  patience: 3
+
+logging:
+  - tensorboard
diff --git a/examples/local/run_diffusion.sh b/examples/local/run_diffusion.sh
@@ -1,14 +1,12 @@
 #!/bin/bash
 
-LOG=debug.log
 CONFIG=config_diffusion.yaml
 DATA_DIR=../../data/si_diffusion_v1
 PROCESSED_DATA=${DATA_DIR}/processed
 DATA_WORK_DIR=./tmp_work_dir/
 OUTPUT=debug
 
 python ../../crystal_diffusion/train_diffusion.py \
-    --log $LOG \
     --config $CONFIG \
     --data $DATA_DIR \
     --processed_datadir $PROCESSED_DATA \