keras model.load_weights 错误 NoneType' 对象没有属性 'fit'

Question

希望有人能提供帮助，我正在尝试基于另一个预训练模型来训练模型，但我不断收到此错误。

提前致谢。 和愉快的编码。

型号

model = Unet(backbone_name=efficientnetb2, encoder_weights='imagenet',
             input_shape=(256, 256, 3),
             classes=1, activation='sigmoid')

    model = get_model(
        mparams['backbone'], 
        input_shape=(mparams['img_size'], mparams['img_size'], 3),
        loss_type=mparams['loss'],
        umodel=mparams['umodel'],
        lr=mparams['lr']
    )
    
    # load model
    model = model.load_weights('./models_v24/model_0.hdf5')

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-19-b3a8ba8f71c6> in <module>
     41             resize=None
     42         )
---> 43         model, history = train_model(PARAMS, n, train_datagen, val_datagen)
     44         plt.plot(history.history['loss'], label='loss')
     45         plt.plot(history.history['val_loss'], label='val_loss')

<ipython-input-17-191deeedc4ed> in train_model(mparams, n_fold, train_datagen, val_datagen)
     53     model = model.load_weights('./models_v24/model_0.hdf5')
     54 
---> 55     history = model.fit(
     56                         train_datagen,
     57                         validation_data=val_datagen,

AttributeError: 'NoneType' object has no attribute 'fit'

Answer 1

您的评论 # load model 不正确。您只是在此处加载权重，不是整个模型。此函数在以 HDF5 格式加载权重时返回 None，这解释了此处的当前错误。

如果创建了模型，你只需要model.load_weights('./models_v24/model_0.hdf5')。那么你应该能够拟合模型。

model = get_model(...)

# load weights
model.load_weights('./models_v24/model_0.hdf5')

我还注意到您不接受 SO 中的某些答案。请看What should I do when someone answers my question?

Answer 2

不要使用 [model = model.load_weights()] 直接使用 [model.load_weights()]