From 3358d96bfea0c7a3021b1c5cf714a2c90141bc1d Mon Sep 17 00:00:00 2001
From: Victor Turrisi <vt.turrisi@gmail.com>
Date: Sun, 10 Oct 2021 18:47:17 -0300
Subject: [PATCH] Fix/bash files (#158)

* Update byol.sh

* Update barlow.sh
---
 bash_files/pretrain/cifar/barlow.sh | 2 ++
 bash_files/pretrain/cifar/byol.sh   | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/bash_files/pretrain/cifar/barlow.sh b/bash_files/pretrain/cifar/barlow.sh
index b0d5c27d..ae6be8a6 100644
--- 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
index 4098aea7..3c7377fe 100644
--- 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 \