ma 0.5

marrlab · Oct 9, 2023 · 643abde · 643abde
1 parent 26c3658
commit 643abde
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/run_fbopt_mnist.sh b/run_fbopt_mnist.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=1 --tr_d 0 3 --task=mnistcolor10 --bs=16 --aname=jigen --trainer=fbopt --nname=conv_bn_pool_2 --epos=500 --es=50 --mu_init=0.00001
+python main_out.py --te_d=1 --tr_d 0 3 --task=mnistcolor10 --bs=16 --aname=jigen --trainer=fbopt --nname=conv_bn_pool_2 --epos=500 --es=50 --mu_init=0.00001 --coeff_ma_output_state=0.5