胶囊网络

胶囊网络 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