pytorch网络模型可视化方法:tensorboardX
步骤:
pip 安装tensorboardX
from tensorboardX import SummaryWriter
关键代码:
with SummaryWriter(comment='tcn') as w:
w.add_graph(model, input_to_model=dummy_input)
在runs所在文件夹下命令行输入:tensorboard --logdir...
原创
2020-03-15 15:48:36 ·
338 阅读 ·
0 评论