Pytorch - pytorch在colab中使用tensorboard

from torch.utils.tensorboard import SummaryWriter

# default `log_dir` is "runs" - we'll be more specific here
writer = SummaryWriter('runs/fashion_mnist_experiment_1')

pytorch在colab中使用tensorboard,一般是使用SummaryWriter類，在啓動tensorboard之前，也就是執行

tensorboard --logdir=runs

之前，還需要在colab中導入tensorboard

%load_ext tensorboard

完整的代碼也就是

from torch.utils.tensorboard import SummaryWriter

# default `log_dir` is "runs" - we'll be more specific here
writer = SummaryWriter('runs/fashion_mnist_experiment_1')

%load_ext tensorboard

tensorboard --logdir=runs

參考

1. https://pytorch.org/tutorials/intermediate/tensorboard_tutorial.html
2. https://zhuanlan.zhihu.com/p/136245396