现在TFlearn每一步都会向tensorboar发送数据,效率很低。它占用了太多的空间,我不需要它。如果是在一个时代的末尾,那就更好了。或者10个时代。
有没有可能在某个地方设置这个?
发布于 2017-12-21 08:29:42
简短的回答是否定的。Tflearn在运行训练操作的同一行中评估汇总操作,请参阅code
_, train_summ_str = self.session.run([self.train, self.summ_op],feed_batch)
这意味着在每个步骤中都会记录训练集的摘要。也就是说,您可以在tflearn.DNN
模型的fit
方法中指定snapshot_step
和snapshot_epoch
选项,这样验证集中的汇总操作每10个时期才会被评估一次。
正如@Keeehi提到的,如果你真的想这样做,你可以在纯Tensorflow中实现它,在Tensorflow中,你可以更好地控制每一步所做的事情。
https://stackoverflow.com/questions/47834768
复制