为什么训练Unet网络的时候使用fit_generator训练时loss根本不下降,但是用fit时loss却会下降

如题,这是目前博主遇到的一个问题,

参数如下

loss不会下降

model.fit_generator(myGene,steps_per_epoch=500,epochs=100,callbacks=[model_checkpoint],shuffle=True)

 

loss会下降

hist=model.fit(imgs_train, labs_train, batch_size=Bch_Sz, nb_epoch=100, initial_epoch=0,  validation_split=0.2, verbose=1, shuffle= True,callbacks=[model_checkpoint])

 

先挖个坑,等以后发现了问题的原因后再来填坑

你可能感兴趣的:(Python)