Update test-weights of NAS-Bench-201
This commit is contained in:
		
							
								
								
									
										20
									
								
								scripts-search/NAS-Bench-201/test-weights.sh
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										20
									
								
								scripts-search/NAS-Bench-201/test-weights.sh
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,20 @@ | ||||
| #!/bin/bash | ||||
| # bash ./scripts-search/NAS-Bench-201/test-weights.sh cifar10-valid 1 1 | ||||
| echo script name: $0 | ||||
| echo $# arguments | ||||
| if [ "$#" -ne 3 ] ;then | ||||
|   echo "Input illegal number of parameters " $# | ||||
|   echo "Need 3 parameters for dataset, use_12_epoch, and use_validation_set" | ||||
|   exit 1 | ||||
| fi | ||||
| if [ "$TORCH_HOME" = "" ]; then | ||||
|   echo "Must set TORCH_HOME envoriment variable for data dir saving" | ||||
|   exit 1 | ||||
| else | ||||
|   echo "TORCH_HOME : $TORCH_HOME" | ||||
| fi | ||||
|  | ||||
| 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 --use_valid $3 | ||||
		Reference in New Issue
	
	Block a user