PyTorch-MNIST-Tutorial/ ├── LICENSE ├── model.pth ├── model.py ├── README.md ├── test.py ├── train.py ├── utils.py ├── demo.py ├── mnist_train.csv └── mnist_test.csv ...
$ python train_source.py --logdir outputs $ python main.py --logdir outputs --trained outputs/best_model.pt --slope 0.2 For training on Office dataset using ResNet-50 $ python ...