Find file History
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.

ResNets on TensorFlow

To train a ResNet, run,

python3 resnet/ --train_data_path=cifar10/data_batch* --log_root=data/resnet20/log_root \
                              --train_dir=data/resnet20/log_root/train --dataset='cifar10' --model=resnet20 \
                              --num_gpus=1 --checkpoint_dir=data/resnet20/checkpoints --data_format=NCHW

To evaluate resulting checkpoints, run,

python3 -i data/resnet20/checkpoints \
                            -c "python3 resnet/ --mode=eval --eval_data_path=cifar10/test_batch.bin --eval_dir=data/resnet20/log_root/eval --dataset='cifar10' --model=resnet20 --num_gpus=1 --eval_batch_count=100 --eval_once=True --data_format=NCHW"

Make sure to first follow the instructions in resnet/ to get necessary data, etc.