修改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 (将#修改为@）