evaluate_imagenet.sh 180 字节
Newer Older
M
minghaoBD 已提交
1 2 3 4 5 6
#!/bin/bash  
export CUDA_VISIBLE_DEVICES=3
export FLAGS_fraction_of_gpu_memory_to_use=0.98
python3.7 evaluate.py \
          --pruned_model="models" \
          --data="imagenet"