From 044833314d2eb0da6c1ccf1daf22bb5b2e47a6fd Mon Sep 17 00:00:00 2001
From: bishe <123456789@163.com>
Date: Sun, 23 Feb 2025 15:58:59 +0800
Subject: [PATCH] add tau in script

---
 scripts/train.sh | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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