Permalink
Switch branches/tags
Nothing to show
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
executable file 25 lines (23 sloc) 679 Bytes
#!/bin/sh
export CUDA_VISIBLE_DEVICES=0
export LD_PRELOAD="/usr/lib/libtcmalloc.so"
checkpoint="./split_2-2-2/model.ckpt-119999"
basemodel="./group_2-2-2/model.ckpt-199999"
output_file="./split_2-2-2/eval-119999.pkl"
#data_dir="./cifar100/train_val_split"
data_dir="/data1/dalgu/cifar100/train_val_split"
python eval.py --checkpoint $checkpoint \
--basemodel $basemodel \
--output_file $output_file \
--data_dir $data_dir \
--batch_size 100 \
--test_iter 100 \
--num_residual_units 2 \
--k 8 \
--ngroups1 2 \
--ngroups2 2 \
--ngroups3 2 \
--gpu_fraction 0.96 \
--display 10 \
#--finetune True \
#--load_last_layers True \