Tensorboard on colab:torch.utils.tensorboard — PyTorch 1.11.0 documentation
◍新增參數查看:
開始前先確保安裝成功:
pip install tensorboardX # install packages
pip install tensorboard
pip install tensorflow
第一步import tensorboardX (如何使用tensorboardX请参考reference):
from tensorboardX import SummaryWriter # import tensorboardX
第二步初始化一个 SummaryWriter实例:
writer = SummaryWriter() # initialize SummaryWriter
第三步:
%load_ext tensorboard.notebook
or (版本差异,自行选择)
%load_ext tensorboard # 目前我colab適用這個
第四步把你模型需要在tensorboard中可视化的参数写入writer,
# writer.add_scalar('圖像名稱', loss/accuracy數值, n_iteration)
writer.add_scalar('Train/Loss', train_loss, epoch)
writer.add_scalar('Train/Accuracy', train_acc, epoch)
writer.add_scalar('Test/Loss', test_loss, epoch)
writer.add_scalar('Test/Accuracy', test_acc, epoch)
第五步:
%tensorboard --logdir ./ # 即可顯示所有scaler(可邊訓練邊顯示)
◍開啟Tensorboard網頁:
CMD
cd Users/Iris/Desktop/BottleneckTransformers-main # cd至runs所在的目錄下
tensorboard --logdir "./runs" # 運行tensorboard
# http://127.0.0.1:6006/
# 至該網址察看結果(若關掉CMD會無法查看網頁)