update
This commit is contained in:
		
							
								
								
									
										38
									
								
								exp_scripts/zerocostpt_nb201_pipeline.sh
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										38
									
								
								exp_scripts/zerocostpt_nb201_pipeline.sh
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,38 @@ | ||||
| #!/bin/bash | ||||
| script_name=`basename "$0"` | ||||
| id=${script_name%.*} | ||||
| dataset=${dataset:-cifar10} | ||||
| seed=${seed:-0} | ||||
| gpu=${gpu:-"auto"} | ||||
| pool_size=${pool_size:-10} | ||||
| batch_size=${batch_size:-256} | ||||
| edge_decision=${edge_decision:-'random'} | ||||
| validate_rounds=${validate_rounds:-100} | ||||
| metric=${metric:-'jacob'} | ||||
| while [ $# -gt 0 ]; do | ||||
|     if [[ $1 == *"--"* ]]; then | ||||
|         param="${1/--/}" | ||||
|         declare $param="$2" | ||||
|         # echo $1 $2 // Optional to see the parameter:value result | ||||
|     fi | ||||
|     shift | ||||
| done | ||||
|  | ||||
| echo 'id:' $id 'seed:' $seed 'dataset:' $dataset | ||||
| echo 'gpu:' $gpu | ||||
|  | ||||
| cd ../nasbench201/ | ||||
| python3 networks_proposal.py \ | ||||
|     --dataset $dataset \ | ||||
|     --save $id --gpu $gpu --seed $seed \ | ||||
|     --edge_decision $edge_decision --proj_crit $metric \ | ||||
|     --batch_size $batch_size\ | ||||
|     --pool_size $pool_size \ | ||||
|  | ||||
| cd ../zerocostnas/ | ||||
| python3 post_validate.py\ | ||||
|     --ckpt_path ../experiments/nas-bench-201/prop-$id-$seed-$pool_size-$metric\ | ||||
|     --save $id --seed $seed --gpu $gpu\ | ||||
|     --edge_decision $edge_decision --proj_crit $metric \ | ||||
|     --batch_size $batch_size\ | ||||
|     --validate_rounds $validate_rounds\ | ||||
		Reference in New Issue
	
	Block a user