胶囊网络 源码

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