TensorBoard:教程Pytorch:模块'tensorflow._api.v2.io.gfile'没有属性'get_filesystem'

时间:2020-04-16 22:57:18

标签: pytorch google-colaboratory tensorboard

我在运行有关带有嵌入https://pytorch.org/tutorials/intermediate/tensorboard_tutorial.html的Pytorch和TensorBoard的教程时遇到问题

我收到此消息

AttributeError                            Traceback (most recent call last)
<ipython-input-10-e0404d94b4cd> in <module>()
     22 writer.add_embedding(features,
     23                     metadata=class_labels,
---> 24                     label_img=images.unsqueeze(1))
     25 writer.close()

/usr/local/lib/python3.6/dist-packages/torch/utils/tensorboard/writer.py in add_embedding(self, mat, metadata, label_img, global_step, tag, metadata_header)
    779         save_path = os.path.join(self._get_file_writer().get_logdir(), subdir)
    780 
--> 781         fs = tf.io.gfile.get_filesystem(save_path)
    782         if fs.exists(save_path):
    783             if fs.isdir(save_path):

我可以看到投影仪

1 个答案:

答案 0 :(得分:1)

尝试一下

import tensorflow as tf
import tensorboard as tb
tf.io.gfile = tb.compat.tensorflow_stub.io.gfile
相关问题