Pytorch使用tensorboard



  • Tensorboard神奇一键应用于Pytorch

    pip install tensorboardX

    from tensorboardX import SummaryWriter
    
    writer = SummaryWriter('Summary/') 
    
    # in training loop
    writer.add_scalar('Train/Loss', loss, num_iteration)
    

    好,已经可以愉快的玩耍了,至于图片啥的,函数都Ok的,其他图后续有需求再去找吧。

    0_1525838243905_a0587751-5a5f-42d6-9c6f-24b4cbf23582-image.png


 

Copyright © 2018 bbs.dian.org.cn All rights reserved.

与 Dian 的连接断开,我们正在尝试重连,请耐心等待