Pytorch_Template:深度学习pytorch框架训练模板-源码

PyTorch Template 本项目是深度学习模型pytorch框架的一个通用训练模板，源自 . 在原项目上按照自己的习惯做了一些修改。 文件夹组织机构说明如下： Mnist分类模型示例 python main.py 训练过程中可以看到保存的模型，文件在experiments/mnist_exp_0/checkpoints下面。 如果想查看模型的效果，可以运行下面的代码。 agent = MnistAgent(config) test_loader = torch.utils.data.DataLoader( datasets.MNIST('data', train=False, transform=transforms.Compose([ transforms.ToTensor(), transforms.Normalize((0.1307,)