胶囊网络
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
2022-03-08 14:09:20
38KB
Python
1