From 9e5b4af94a6aabbd7dd29d256701588886d01c62 Mon Sep 17 00:00:00 2001 From: smilesun Date: Tue, 10 Oct 2023 11:17:18 +0200 Subject: [PATCH] . --- run_fbopt.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/run_fbopt.sh b/run_fbopt.sh index ed7873414..33226f290 100644 --- 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