/tensorflow_1.x

Primary LanguageJupyter NotebookMIT LicenseMIT

ResNet-Tensorflow

Simple Tensorflow implementation of pre-activation ResNet18, ResNet34, ResNet50, ResNet101, ResNet152

Summary

dataset

  • tiny_imagenet
  • cifar10, cifar100, mnist, fashion-mnist in keras (pip install keras)

Train

  • python main.py --phase train --dataset tiny --res_n 18 --lr 0.1

Test

  • python main.py --phase test --dataset tiny --res_n 18 --lr 0.1

Related works

Author

Junho Kim