Keras
Donald Su
公众号:"Donald Su",微:sug2010
展开
-
利用Keras实现FGSM算法
最近项目中需要使用FGSM攻击方法,使用主流的foolbox和cleverhans中的攻击方法时,发现模型不匹配,所以不能继续使用,需要自己实现一个。原创 2019-07-02 16:36:34 · 4822 阅读 · 27 评论 -
Keras fit函数中传入回调函数,提示『TypeError: set_model() missing 1 required positional argument: 'model'』错误
文章目录问题描述Solutionother问题描述fit中使用回调函数,提示 TypeError: set_model() missing 1 required positional argument: 'model'自己的代码:class TimeHistory(keras.callbacks.Callback): def on_train_begin(self, logs={...原创 2019-07-19 19:11:09 · 5440 阅读 · 0 评论 -
Keras中进行模型训练时,计算每一步迭代所花费时间的方法
在keras中训练模型的时候,准备比较下,有无GPU的模型训练时间花费具体有多大,所以采用了在fit前后调用time.time()的方式计算。代码为:import timestart = time.clock() history = model.fit(train_img, train_labels, epochs=10, batch_size=128)print(time...原创 2019-07-28 21:47:47 · 10658 阅读 · 8 评论