摘要: 看到pytorch居然自带了瓶颈检查的工具:torch.utils.bottleneck 用法: python -m torch.utils.bottleneck 待测脚本路径 例如你原来执行的是 python train.py,现在改成 python -m torch.utils.bottlene 阅读全文
posted @ 2021-11-21 23:50 Rogn 阅读(808) 评论(0) 推荐(0)