pytorch 反卷积 可视化_Toybrick

我的是直接转换pt文件。

D RKNN output shape(batchnormalize): (0 16 16 320)

D Process convolution_at_input136.1_36 ...

D RKNN output shape(convolution): (0 16 16 1280)

D Process batch_norm_at_input137.1_35 ...

D RKNN output shape(batchnormalize): (0 16 16 1280)

D Process hardtanh__at_input138.1_34 ...

D RKNN output shape(relun): (0 16 16 1280)

D Process convolution_at_input139.1_33 ...

D RKNN output shape(convolution): (0 8 8 1280)

D Process batch_norm_at_input140.1_32 ...

D RKNN output shape(batchnormalize): (0 8 8 1280)

D Process relu__at_input141.1_31 ...

D RKNN output shape(relu): (0 8 8 1280)

D Process convolution_at_input142.1_30 ...

D RKNN output shape(convolution): (0 4 4 16)

D Process batch_norm_at_input143.1_29 ...

D RKNN output shape(batchnormalize): (0 4 4 16)

D Process relu__at_input144.1_28 ...

D RKNN output shape(relu): (0 4 4 16)

D Process convolution_at_input145.1_27 ...

D RKNN output shape(convolution): (0 2 2 16)

D Process batch_norm_at_input146.1_26 ...

D RKNN output shape(batchnormalize): (0 2 2 16)

D Process relu__at_input147.1_24 ...

D RKNN output shape(relu): (0 2 2 16)

D Process convolution_at_input148.1_18 ...

D RKNN output shape(convolution): (0 2 2 64)

D Process relu__at_input149.1_12 ...

D RKNN output shape(relu): (0 2 2 64)

D Process convolution_at_hm_x.1_6 ...

D Process output_of_convolution_at_2269_5 ...

D RKNN output shape(output): (0 2 2 2)

I Build mobilenet1 complete.

D Optimizing network with force_1d_tensor, swapper, merge_layer, auto_fill_bn, resize_nearest_transformer, auto_fill_multiply, merge_avgpool_conv1x1, auto_fill_zero_bias, proposal_opt_import

D Merge ['convolution_at_input145.1_27', 'batch_norm_at_input146.1_26'] (convolution)

D Merge ['convolution_at_input142.1_30', 'batch_norm_at_input143.1_29'] (convolution)

D Merge ['convolution_at_input139.1_33', 'batch_norm_at_input140.1_32'] (convolution)

E Catch exception when loading pytorch model: /Users/gyq/DeepLearning/CenterNet_in_mac/mobilenet1.pt!

E Traceback (most recent call last):

E   File "rknn/api/rknn_base.py", line 567, in rknn.api.rknn_base.RKNNBase.load_pytorch

Load pytorch model failed!

E   File "rknn/base/RKNNlib/app/importer/import_pytorch.py", line 121, in rknn.base.RKNNlib.app.importer.import_pytorch.ImportPytorch.run

E   File "rknn/base/RKNNlib/app/helper/mergehelper.py", line 155, in rknn.base.RKNNlib.app.helper.mergehelper.MergeHelper.merge

E   File "rknn/base/RKNNlib/optimize/optimizer.py", line 299, in rknn.base.RKNNlib.optimize.optimizer.Optimizer.apply

E   File "rknn/base/RKNNlib/optimize/rules/merge_layer.py", line 81, in rknn.base.RKNNlib.optimize.rules.merge_layer.MergeLayer.apply

E   File "rknn/base/RKNNlib/optimize/rules/merge_layer.py", line 105, in rknn.base.RKNNlib.optimize.rules.merge_layer.MergeLayer._loop

E   File "rknn/base/RKNNlib/optimize/rules/merge_layer_ext_proc.py", line 86, in rknn.base.RKNNlib.optimize.rules.merge_layer_ext_proc.m_l1_bn

E ValueError: operands could not be broadcast together with shapes (4,4,16,1280) (16,)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值