From 0fc35851af9bff720439a2f01aec6223bf17eae0 Mon Sep 17 00:00:00 2001 From: smilesun Date: Mon, 2 Oct 2023 14:52:07 +0200 Subject: [PATCH] . --- run_fbopt_pacs.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/run_fbopt_pacs.sh b/run_fbopt_pacs.sh index 73c1e61ab..81e7a1189 100644 --- 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