-
深度学习PyTorch框架训练的模板：Pytorch_Template
资源介绍
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,)