摘要:
前言 pin_memory 和 non_blocking的作用分别是什么?网上看了很多解释,只是稀里糊涂的有个感觉,就是用了这玩意速度能变快,但是不知所以然,这篇文章希望能帮助你解惑,也给自己做个笔记,以备日后查阅。 train_sampler = None train_loader = torch 阅读全文
posted @ 2022-03-08 23:43
marsggbo
阅读(4713)
评论(0)
推荐(1)
摘要:
参考: https://www.liujiangblog.com/course/python/83 阅读全文
posted @ 2022-03-08 11:55
marsggbo
阅读(71)
评论(0)
推荐(0)