Open cbwces opened 3 years ago
修改Caffe/caffe.proto文件并生成新的caffe_pb2.py,修复pooling层转化后，caffe无法读取ceil_mode参数问题 添加avgpool层return_indices参数以修复AvgPool2d在正向传递时参数个数不足的错误 修改example/resnet_pytorch_2_caffe.py，使用model/resnet.py中的模型而不是torchvision中的resnet模型，将model/resnet.py中ResNet类使用的AvgPool2d替换为AdaptiveAvgPool2d，flatten算子替换为view
Caffe/caffe.proto
caffe_pb2.py
example/resnet_pytorch_2_caffe.py
model/resnet.py
修改
Caffe/caffe.proto文件并生成新的caffe_pb2.py,修复pooling层转化后，caffe无法读取ceil_mode参数问题 添加avgpool层return_indices参数以修复AvgPool2d在正向传递时参数个数不足的错误 修改example/resnet_pytorch_2_caffe.py，使用model/resnet.py中的模型而不是torchvision中的resnet模型，将model/resnet.py中ResNet类使用的AvgPool2d替换为AdaptiveAvgPool2d，flatten算子替换为view