如何解决修改Tensorboard类时,模块'tensorflow._api.v2.summary'没有'FileWriter'成员错误
我试图修改Tensorboard
类以拥有一个日志文件,因为我需要多次适应(我正在使用深度强化学习)。但是我有以下错误:
Module 'tensorflow._api.v2.summary' has no 'FileWriter' member
Module 'tensorboard.summary._tf.summary' has no 'FileWriter' member
Instance of 'ModifiedTensorBoard' has no '_write_logs' member
这是代码:
from keras.callbacks import TensorBoard
import tensorflow as tf
# Own Tensorboard class
class ModifiedTensorBoard(TensorBoard):
# Overriding init to set initial step and writer (we want one log file for all .fit() calls),since we have to fit multiple times
def __init__(self,**kwargs):
super().__init__(**kwargs)
self.step = 1
self.writer = tf.summary.FileWriter(self.log_dir)
# Overriding this method to stop creating default log writer
def set_model(self,model):
pass
# Overrided,saves logs with our step number
# (otherwise every .fit() will start writing from 0th step)
def on_epoch_end(self,epoch,logs=None):
self.update_stats(**logs)
# Overrided
# We train for one batch only,no need to save anything at epoch end
def on_batch_end(self,batch,logs=None):
pass
# Overrided,so won't close writer
def on_train_end(self,_):
pass
# Custom method for saving own metrics
# Creates writer,writes custom metrics and closes writer
def update_stats(self,**stats):
self._write_logs(stats,self.step)
我将不胜感激!谢谢。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。