.

marrlab · Oct 2, 2023 · 0fc3585 · 0fc3585
1 parent a9f1fc8
commit 0fc3585
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/run_fbopt_pacs.sh b/run_fbopt_pacs.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=sketch --tpath=examples/tasks/task_pacs_path_list.py --bs=16 --aname=jigen --trainer=fbopt --nname=alexnet --epos=20
+python main_out.py --te_d=sketch --tpath=examples/tasks/task_pacs_path_list.py --bs=16 --aname=jigen --trainer=fbopt --nname=alexnet --epos=100 --es=100