随笔分类 - Keras
摘要:1、使用predict时,需设置batch_size 查看keras文档中,predict函数原型:predict(self, x, batch_size=32, verbose=0)说明:只使用batch_size=32,也就是说每次将batch_size=32的数据通过PCI总线传到GPU,然后
阅读全文
摘要:1.使用tensorboard可视化ACC,loss等曲线 然后新开一个终端 输入: 之后打开终端给出的网址即可。 2.直接使用matplotlib画出训练LOSS与ACC曲线 第一步: 第二步: 第三步:
阅读全文
摘要:训练好的模型,想要输入中间层的特征图,有两种方式: 1. 通过model.get_layer的方式。创建新的模型,输出为你要的层的名字。 创建模型,debug状态可以看到模型中,base_model/layers,图中红框即为layer名字,根据你想输出的层填写。最后网络feed数据后,输出的就是中
阅读全文

浙公网安备 33010602011771号