bubbliiiing / efficientdet-pytorch

这是一个efficientdet-pytorch的源码,可以用于训练自己的模型。
MIT License
305 stars 61 forks source link

train.py #55

Open MoonLooking opened 9 months ago

MoonLooking commented 9 months ago

Fail To Load Key: ['classifier.header.pointwise_conv.conv.weight', 'classifier.header.pointwise_conv.conv.bias'] …… Fail To Load Key num: 2

温馨提示,head部分没有载入是正常现象,Backbone部分没有载入是错误的。 The expanded size of the tensor (46917) must match the existing size (49104) at non-singleton dimension 1. Target sizes: [1, 46917, 4]. Tensor sizes: [1, 49104, 4] Error occurs, No graph saved 在运行时出现了这个错误