Визуализация моделей, данных и обучение с помощью тензорной доски в pytorch

Question

После этого урока для визуализации изображений в тензорной доске с использованием torch.utils.tensorboard получена ошибка

from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter('runs/fashion_mnist_experiment_1')
writer.add_embedding(features,
                metadata=class_labels,
                label_img=images.unsqueeze(1))
writer.close()

Ошибка:

        AttributeError: module 'tensorflow_core._api.v2.io.gfile' has no attribute 'get_filesystem'

Answer 1

попробуйте добавить

writer = SummaryWriter(path)
img_grid = torchvision.utils.make_grid(torch.FloatTensor(imgs)) #imgs represent a batch of images with shape BxCxHxW
writer_add_image('images', img_grid)

источник: https://pytorch.org/tutorials/intermediate/tensorboard_tutorial.html#writing к тензорной доске