Pytorch损失曲线

安装tensorboardX

Ubuntu系统这个文件保存了训练时的损失
在这个文件夹打开终端
输入tensorboard --logdir=./