Keras多输出(多任务)如何设置fit_generator
在使用Keras的时候,因为需要考虑到效率问题,需要修改fit_generator来适应多输出# create modelmodel = Model(inputs=x_inp, outputs=[main_pred, aux_pred])# complie modelmodel.compile( optimizer=optimizers.Adam(lr=learning_rate)...
原创
2019-12-10 21:44:04 ·
3026 阅读 ·
1 评论