【文件属性】：
文件名称：Pytorch_Template:深度学习pytorch框架训练模板
文件大小：361KB
文件格式：ZIP
更新时间：2021-04-14 03:40:49
Python
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,)
【文件预览】：
Pytorch_Template-master
----.gitignore(1KB)
----README.md(1KB)
----utils()
--------misc.py(1KB)
--------config.py(4KB)
--------replay_memory.py(896B)
--------metrics.py(4KB)
--------my_aug.py(3KB)
--------generate_class_weights.py(8KB)
--------env_utils.py(2KB)
--------voc_utils.py(4KB)
--------__init__.py(411B)
--------dirs.py(584B)
--------train_utils.py(740B)
----configs()
--------mnist_exp_0.json(636B)
--------dcgan_exp_0.json(739B)
--------condensenet_exp_0.json(956B)
--------erfnet_exp_0.json(839B)
--------dqn_exp_0.json(726B)
--------example_exp_0.json(515B)
----LICENSE(1KB)
----agents()
--------base.py(2KB)
--------example.py(3KB)
--------dqn.py(9KB)
--------mnist.py(5KB)
--------dcgan.py(9KB)
--------erfnet.py(10KB)
--------__init__.py(30B)
--------condensenet.py(9KB)
----main.py(310B)
----requirements.txt(0B)
----graphs()
--------backbone()
--------models()
--------weights_initializer.py(1KB)
--------__init__.py(393B)
--------losses()
--------neck()
----tutorials()
--------Mnist()
----demo.py(833B)
----data()
--------assets()
----data_loader()
--------coco2017.py(7KB)
--------example.py(3KB)
--------cifar10.py(4KB)
--------mnist.py(4KB)
--------__init__.py(394B)
--------celebA.py(3KB)
--------voc2012.py(6KB)