iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix the potential memory leak in NAS-Bench-201 clear_param

Browse Source
This commit is contained in:
D-X-Y
2020-03-21 01:33:07 -07:00
parent b702ddf5a2
commit 22025887f1
9 changed files with 40 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts-search/NAS-Bench-201/test-weights.sh
View File

@@ -15,7 +15,7 @@ else
echo "TORCH_HOME : $TORCH_HOME"
fi
OMP_NUM_THREADS=4 python exps/NAS-Bench-201/test-weights.py \
CUDA_VISIBLE_DEVICES='' OMP_NUM_THREADS=4 python exps/NAS-Bench-201/test-weights.py \
--base_path $HOME/.torch/NAS-Bench-201-v1_1-096897 \
--dataset $1 \
--use_12 $2