Benchmark code
Clone or download
Pull request Compare This branch is 51 commits ahead, 398 commits behind tensorflow:master.
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 for ImageNet on TensorFlow

To train a ResNet, run,

python3 --model=resnet50 --data_dir=../../../../data/imagenet/
                             --num_batches=1800000 --subset=train --learning_rate=0.05
                             --learning_rate_decay_factor=0.1 --num_epochs_per_decay=30
                             --optimizer=momentum --weight_decay=0.0001

This command produces model checkpoints written after every epoch. To evaluate each of these checkpoints, run,

python3 -i /lfs/1/deepak/checkpoints/resnet50_lr=0.05/
                            -c "python --model=resnet50 --eval --data_dir=/lfs/1/deepak/data/imagenet/ --eval_subset=validation --num_batches=100 --batch_size=500"

Other example command lines are available in the scripts/ directory (for example, training and evaluating ResNet152 on 4 GPUs).

Make sure to first follow the instructions in the TensorFlow models repository to get necessary data, etc.