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