From d4ba5d8114ca95ab95c4935ddf951113456c152d Mon Sep 17 00:00:00 2001
From: bishe <123456789@163.com>
Date: Sun, 23 Feb 2025 16:02:17 +0800
Subject: [PATCH] add num_timesteps

---
 checkpoints/ROMA_UNSB_001/loss_log.txt        |   1 +
 checkpoints/ROMA_UNSB_001/train_opt.txt       |   1 +
 .../roma_unsb_model.cpython-39.pyc            | Bin 19348 -> 19389 bytes
 models/roma_unsb_model.py                     |   1 +
 scripts/train.sh                              |   3 ++-
 5 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/checkpoints/ROMA_UNSB_001/loss_log.txt b/checkpoints/ROMA_UNSB_001/loss_log.txt
index 485f77f..a6f9e99 100644
--- a/checkpoints/ROMA_UNSB_001/loss_log.txt
+++ b/checkpoints/ROMA_UNSB_001/loss_log.txt
@@ -1,2 +1,3 @@
 ================ Training Loss (Sun Feb 23 15:46:44 2025) ================
 ================ Training Loss (Sun Feb 23 15:52:29 2025) ================
+================ Training Loss (Sun Feb 23 16:00:07 2025) ================
diff --git a/checkpoints/ROMA_UNSB_001/train_opt.txt b/checkpoints/ROMA_UNSB_001/train_opt.txt
index 0658b40..18c3b2c 100644
--- a/checkpoints/ROMA_UNSB_001/train_opt.txt
+++ b/checkpoints/ROMA_UNSB_001/train_opt.txt
@@ -76,6 +76,7 @@ nce_includes_all_negatives_from_minibatch: False
            serial_batches: False                         
 stylegan2_G_num_downsampling: 1                             
                    suffix:                               
+                      tau: 0.1                           
          update_html_freq: 1000                          
                   use_idt: False                         
                   verbose: False                         
diff --git a/models/__pycache__/roma_unsb_model.cpython-39.pyc b/models/__pycache__/roma_unsb_model.cpython-39.pyc
index 277be8fe6c2823d02c7799aeec359311a65fa7e5..1b4d9539fcb1c3b40418c7b8435bb2b4a7ef46d9 100644
GIT binary patch
delta 253
zcmbO-opJAUM&3kTUM>b8Sa4-m`udH$KH`i8o1?_77_Avn^hy|0I8yXeSbA9)8EP1^
z*isBq40{<t>;>!#8Df}fnQEEA;(RGgDMr0aAemqWP2<gbCHh!It5|h)OA<?~cuI>?
zQxr1u6iV}olO}u19%D3{EF@>Z%p}CL*;(!oGvnvUY)VTQe{Ei^^oxn{_2y-&W=xDq
zoA0U%Ffm$eey6#Xi7|5X9GzuMjOm-5^<@|t>o+GDq;W9LoqXJ4C*$+Yjh4F1jNO~p
y+bA+Iif_JX*UiYNx;fV2E;FP4W^0#nX2!71o82EWGFDBV>Uok;V{^QhDI)*_3r(H?

delta 191
zcmV;w06_n}mjRTQ0S#dd4GI7N006bqx@TOm4NMmSUb9*kIspX+WG-YcldBjR3owlW
z0RVgeQvfM3v(Ffg1e0<d#sM^wIUX<p1G8)%!UF;FlP@Bq0r|7kBKiUW>a))!HUa@5
zvj!*-0s%L(7b>j+0a&xeET#ehX0vQB9037-vxzWf2mzdv@Hn{v=Ci6fEdv3Hv)DW#
t0s$7Y`aX*R0VK1AK-~iYF0*7rcLM=cv)o7F0RefFzDdaeC$or3H32BkKB)iz

diff --git a/models/roma_unsb_model.py b/models/roma_unsb_model.py
index 4efe49e..691121d 100644
--- a/models/roma_unsb_model.py
+++ b/models/roma_unsb_model.py
@@ -222,6 +222,7 @@ class RomaUnsbModel(BaseModel):
         parser.add_argument('--atten_layers', type=str, default='1,3,5', help='compute Cross-Similarity on which layers')
         
         parser.add_argument('--tau', type=float, default=0.1, help='used in unsb')
+        parser.add_argument('--num_timesteps', type=int, default=10, help='used in unsb')
 
         parser.set_defaults(pool_size=0)  # no image pooling
 
diff --git a/scripts/train.sh b/scripts/train.sh
index c0a4781..d005fe5 100755
--- a/scripts/train.sh
+++ b/scripts/train.sh
@@ -28,4 +28,5 @@ python train.py \
     --num_patches 256 \
     --flip_equivariance False \
     --eta_ratio 0.1 \
-    --tau 0.1
+    --tau 0.1 \
+    --num_timesteps 10