【文件属性】：
文件名称：TensorMONK:深度学习模型的集合（PyTorch实现）
文件大小：13.83MB
文件格式：ZIP
更新时间：2021-05-09 18:58:15
pytorch vae densenet resnet unet
TensorMONK
深度学习架构的集合（PyTorch实现）。
依存关系
python> = 3.6
PyTorch> 1.3
火炬视觉
视觉
2012 ImageNet识别任务的训练模型
如果您有更多的nvidia卡和内核可用，请在./ImageNet.sh中相应地调整批处理大小（BSZ），GPU数量（gpus）和线程数量（cpus）。 接下来，选择一个可用的体系结构并更新您的训练和验证文件夹位置（trainDataPath和testDataPath）。 最后，运行./ImageNet.sh。
在MNIST上培训CapsuleNet
要在MNIST上复制Hinton的论文，请运行以下命令：
python Capsule.py -A胶囊-B 256 -E 500 --optimizer亚当--gpus 2 --cpus 6 --trainDataPath ./data --te