diff --git a/scripts/train.sh b/scripts/train.sh
index 3315465..c0a4781 100755
--- a/scripts/train.sh
+++ b/scripts/train.sh
@@ -27,4 +27,5 @@ python train.py \
     --lmda_1 0.1 \
     --num_patches 256 \
     --flip_equivariance False \
-    --eta_ratio 0.1
+    --eta_ratio 0.1 \
+    --tau 0.1