chenghtao的博客

问题描述：建立好model之后，用model.fit()函数进行训练，发现超出显存容量(一共有12G) 问题分析：fit()函数训练时，将全部训练集载入显存之后，才开始分批训练。显然很容易就超出了12G 解决办法：用fit_generator函数进行训练fit_generator函数将训练集分批载入显存，但需要自定义其第一个参数——generator函数，从而分批将训练集送入显存def data

问题描述：训练结束后，保存model为hdf5和yaml格式的文件yamlFilename = os.path.join(dir,filename)yamlModel = model.toyaml()with open(yamlFilename, "w") as yamlFile: yamlFile.write(yamlModel)随后load modelwith open(chkptF

问题描述:如何将建立好的keras模型方便地绘制成图形? 解决办法:利用keras中的plot_model方法,但该方法需要一大堆依赖包,需要一个一个逐次安装from keras.utils.vis_utils import plot_modelplot_model(model, to_file="model.png",show_shapes=True);安装过程: 1.安装gra