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