当新模型的结构与预训练的模型结构不同,如当fine tuning或调参时,载入keras模型,可通过设置by_name=True:
basemodel.load_weights('../pretrain-models/keras.hdf5',by_name=True)
model.fit_generator中steps_per_epoch参数表示每个epoch的batch数,即等于num_dataset/batch_size
待续。。。
当新模型的结构与预训练的模型结构不同,如当fine tuning或调参时,载入keras模型,可通过设置by_name=True:
basemodel.load_weights('../pretrain-models/keras.hdf5',by_name=True)
model.fit_generator中steps_per_epoch参数表示每个epoch的batch数,即等于num_dataset/batch_size
待续。。。