文件名称：胶囊网络
文件大小：38KB
文件格式：ZIP
更新时间：2024-03-17 05:55:25
Python
胶囊网络 PyTorch实现以下文件： Sara Sabour，Nicholas Frosst和Geoffrey Hinton的 正式执行 Sara Sabour的（TensorFlow） 视觉表现 图片来源： Mike Ross ， 运行实验 有关详细信息，请运行python main.py --help 重构图像与原始图像的示例 要求： PyTorch（ ） NumPy（ ） 显卡 默认的超参数（类似于纸张）： 每个GPU batch_size = 128 初始learning_rate = 0.001 指数lr_decay = 0.96 路由迭代次数（ num_routing ）= 3 损失函数超参数（请参阅 ）： 保证金亏损的Lambda = 0.5 重建损失的比例因子= 0.0005 GPU Speed基准： （具有上述超参数） 单个GeForce
【文件预览】：
capsule-network-master
----capsules.py(4KB)
----main.py(3KB)
----model.py(2KB)
----loss.py(2KB)
----LICENSE(1KB)
----README.md(2KB)
----.gitignore(168B)
----reconstructed.png(30KB)
----trainer.py(4KB)