rotta

LINs-lab · Apr 6, 2024 · 19cb3ce · 19cb3ce
1 parent f8d09ab
commit 19cb3ce
Show file tree

Hide file tree

Showing 5 changed files with 678 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -12,6 +12,7 @@ data/
 logs/
 # pretrain/
 runs/
+todo/
 *.out
 *.pth
 *.sh

diff --git a/ttab/configs/algorithms.py b/ttab/configs/algorithms.py
@@ -92,4 +92,13 @@
         * 0.40,  # The threshold for reliable minimization in SAR.
         "reset_constant_em": 0.2,  # threshold e_m for model recovery scheme
     },
+    "rotta":{
+        "optimizer": "Adam",
+        "nu": 0.001,
+        "memory_size": 64,
+        "update_frequency": 64,
+        "lambda_t": 1.0,
+        "lambda_u": 1.0,
+        "alpha": 0.05,
+    }
 }
diff --git a/ttab/model_adaptation/__init__.py b/ttab/model_adaptation/__init__.py
@@ -12,6 +12,7 @@
 from .tent import TENT
 from .ttt import TTT
 from .ttt_plus_plus import TTTPlusPlus
+from .rotta import Rotta
 
 
 def get_model_adaptation_method(adaptation_name):
@@ -29,4 +30,5 @@ def get_model_adaptation_method(adaptation_name):
         "conjugate_pl": ConjugatePL,
         "cotta": CoTTA,
         "eata": EATA,
+        "rotta": Rotta,
     }[adaptation_name]
-Original file line number
+Diff line change
@@ Expand Up / @@ -12,6 +12,7 @@ data/ @@
     logs/
     # pretrain/
     runs/
+    todo/
     *.out
     *.pth
     *.sh
@@ Expand Down @@