2023年5月23日
升级到PyTorch 2.0的技巧总结
摘要： PyTorch 2.0 发布也有一段时间了，大家是不是已经开始用了呢？ PyTorch 2.0 通过引入 torch.compile，可以显着提高训练和推理速度。 与 eagerly 模式相反，编译 API 将模型转换为中间计算图（FX graph），然后以某种方式将其编译为低级计算内核，这样可以提 阅读全文
posted @ 2023-05-23 10:42 deephub 阅读(63) 评论(0) 推荐(0) 编辑