jiahongchen / resnet-pytorch Goto Github PK
View Code? Open in Web Editor NEWThis repo replicates the ResNet on MNIST/FashionMNIST dataset, using PyTorch torchvision model. The torchvision model is reused by splitting the ResNet into a feature extractor and a classifier. And the training is conducted with/without the pre-trained model.
License: Apache License 2.0