Merge branch 'fbopt' into fbopt_setpoint_ma

marrlab · Oct 10, 2023 · 382b8b9 · 382b8b9
2 parents ee78952 + 9e5b4af
commit 382b8b9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/run_fbopt.sh b/run_fbopt.sh
@@ -3,4 +3,4 @@
 # although garbage collector has been explicitly called, sometimes there is still CUDA out of memory error
 # so it is better not to use GPU to do the pytest to ensure every time there is no CUDA out of memory error occuring
 # pytest -s tests/test_fbopt.py
-python main_out.py --te_d=caltech --task=mini_vlcs --bs=16 --aname=jigen --trainer=fbopt --nname=alexnet --epos=200 --es=100 --init_mu=1.0
+python main_out.py --te_d=caltech --task=mini_vlcs --bs=16 --aname=jigen --trainer=fbopt --nname=alexnet --epos=200 --es=5 --mu_init=1.0