add tau in script

2025-02-23 15:58:59 +08:00 · 2025-02-23 15:58:59 +08:00 · 044833314d
commit 044833314d
parent 7b9e3003f1
1 changed files with 2 additions and 1 deletions
--- 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