IndexError: вывод списка индексов за пределы диапазона, когда я пытаюсь использовать load_weights () для загрузки моих обученных весов моей собственной модели

Question

Я создаю небольшой ResNet, обучаю и оцениваю его успешно, когда я не использовал load_weights. И сохраняю вес с помощью modelCheckPoint.

Код работает правильно, когда я не использую функцию load_weights (),Но ошибка IndexError возникает, когда у меня есть эта строка.

print("***** conv_model *****")
model=A2OD_N(c=len_closeness,p=len_period,
             T=T,F=F,H=H,W=W,
             pre_F=64,R_N=2,
             CF=64,
             drop=0,
             kernel1=1,
             is_summary=True,
             nb_residual_unit=3)

file_conv='MODEL/pop2flow_'+str(iterate)+'.hdf5'
if train:
    model_checkpoint=ModelCheckpoint(
            filepath=file_conv,
            monitor='val_rmse',
            verbose=1,
            save_best_only=True,
            save_weights_only=True,
            mode='min',
            period=1
        )
    print('=' * 10)
    print("***** training conv_model *****")
    history = model.fit(X_train, Y_train,
                        epochs=epoch,
                        batch_size=batch_size,
                        validation_split=0.1,
                        callbacks=[model_checkpoint],
                        verbose=1)
print('=' * 10)
print('***** evaluate *****')
model.load_weights(file_conv)

***** оценка *****

Traceback (most recent call last):
  File "executor.py", line 78, in <module>
    model.load_weights(file_conv)
  File "/home/loooffeeeey/Documents/pop2flow/pop2flow/lib/python3.6/site-packages/keras/engine/network.py", line 1166, in load_weights
    f, self.layers, reshape=reshape)
  File "/home/loooffeeeey/Documents/pop2flow/pop2flow/lib/python3.6/site-packages/keras/engine/saving.py", line 1045, in load_weights_from_hdf5_group
    reshape=reshape)
  File "/home/loooffeeeey/Documents/pop2flow/pop2flow/lib/python3.6/site-packages/keras/engine/saving.py", line 682, in preprocess_weights_for_loading
    weights = convert_nested_model(weights)
  File "/home/loooffeeeey/Documents/pop2flow/pop2flow/lib/python3.6/site-packages/keras/engine/saving.py", line 670, in convert_nested_model
    original_backend=original_backend))
  File "/home/loooffeeeey/Documents/pop2flow/pop2flow/lib/python3.6/site-packages/keras/engine/saving.py", line 682, in preprocess_weights_for_loading
    weights = convert_nested_model(weights)
  File "/home/loooffeeeey/Documents/pop2flow/pop2flow/lib/python3.6/site-packages/keras/engine/saving.py", line 658, in convert_nested_model
    original_backend=original_backend))
  File "/home/loooffeeeey/Documents/pop2flow/pop2flow/lib/python3.6/site-packages/keras/engine/saving.py", line 800, in preprocess_weights_for_loading
    elif layer_weights_shape != weights[0].shape:
IndexError: list index out of range