我正在尝试使用fit_generator
带有keras的模型来tensorboard_callback
分析特定时期。我正在为生成器运行以下代码:
def gen(source):
loopable = iter(source)
for batch in loopable:
yield (batch[0], batch[1])
在主要的训练脚本中,我将实例化生成器并将模型与张量板回调一起使用,如下所示:
train_gen = gen(train_datasource)
log_dir="logs/profile/" + datetime.datetime.now().strftime("%Y%m%d-%H%M%S")
tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir=log_dir, histogram_freq=1, profile_batch = 3)
m.fit_generator(train_gen, epochs=5, steps_per_epoch=500, use_multiprocessing=True, workers=32, callbacks=[tensorboard_callback])
我面临的主要问题是,在出现'Function' has no attribute 'fetch_callbacks'
以下堆栈跟踪信息时,培训总是因错误而中止:
m.fit_generator(train_gen, epochs=5, steps_per_epoch=500, use_multiprocessing=True, workers=32, callbacks=[tensorboard_callback])
File "/usr/local/lib/python2.7/dist-packages/keras/legacy/interfaces.py", line 91, in wrapper
return func(*args, **kwargs)
File "/usr/local/lib/python2.7/dist-packages/keras/engine/training.py", line 1418, in fit_generator
initial_epoch=initial_epoch)
File "/usr/local/lib/python2.7/dist-packages/keras/engine/training_generator.py", line 177, in fit_generator
callbacks.on_epoch_begin(epoch)
File "/usr/local/lib/python2.7/dist-packages/keras/callbacks.py", line 65, in on_epoch_begin
callback.on_epoch_begin(epoch, logs)
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/keras/callbacks_v1.py", line 386, in on_epoch_begin
self.merged] = self._fetch_callback
AttributeError: 'Function' object has no attribute 'fetch_callbacks'
我正在使用tensorflow 1.15,也尝试降级到1.14,但仍然没有成功。我正在尝试使用张量板回调来调试除第一个以外的特定时期的性能。但是到目前为止,我的尝试未能使回调函数正确运行。我确保GPU也在运行并正确检测到。
任何帮助将非常感激。
我最终使用了tf.keras
fit函数而不是fit generator,它按预期正常工作:
m.fit(x=train_gen, epochs=5, steps_per_epoch=500, use_multiprocessing=True, workers=8, callbacks=[tensorboard_callback])
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句