Fix/bash files (#158)

* Update byol.sh * Update barlow.sh
vturrisi · Oct 10, 2021 · 3358d96 · 3358d96
1 parent 555177a
commit 3358d96
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/bash_files/pretrain/cifar/barlow.sh b/bash_files/pretrain/cifar/barlow.sh
@@ -26,6 +26,8 @@ python3 ../../../main_pretrain.py \
     --name barlow-$1 \
     --project solo-learn \
     --entity unitn-mhug \
+    --wandb \
+    --save_checkpoint \
     --method barlow_twins \
     --proj_hidden_dim 2048 \
     --proj_output_dim 2048 \

diff --git a/bash_files/pretrain/cifar/byol.sh b/bash_files/pretrain/cifar/byol.sh
@@ -27,6 +27,8 @@ python3 ../../../main_pretrain.py \
     --name byol-$1 \
     --project solo-learn \
     --entity unitn-mhug \
+    --wandb \
+    --save_checkpoint \
     --method byol \
     --proj_output_dim 256 \
     --proj_hidden_dim 4096 \