0
点赞
收藏
分享

微信扫一扫

Tensorboard on colab

爱情锦囊 2022-04-01 阅读 77
python

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會無法查看網頁)

举报

相关推荐

0 条评论