文件名称：TensorMONK:深度学习模型的集合（PyTorch实现）
文件大小：13.83MB
文件格式：ZIP
更新时间：2024-05-29 12:44:55
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