Skip to content

ResNet for CIFAR with Estimator API and tf.keras.Model class

Notifications You must be signed in to change notification settings

mmkamani7/ResNet-Cifar

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ResNet-Cifar

ResNet for CIFAR with Estimator API and tf.keras.Model class

You can download each dataset and convert it to tfrecords using:

python generate_cifar_tfrecords.py --data-dir=./cifar10 --dataset cifar10

Then, you can run the ResNet model using:

python main.py --data-dir=./cifar10 \
               --job-dir=./log/cifar10 \
               --dataset cifa10 \
               --num-layers=20 \
               --version v1