2023年4月5日
关于pytorch使用时的一点心得，主要是关于tensor的设备管理问题
摘要： 在研究论文代码的过程中，pytorch中的模型通常是首先在__init__方法中创建tensor变量以及一些子模型 第一个冷知识：在 PyTorch 中，Tensor 的初始化是在 CPU 上进行的，这是因为大多数计算机都有 CPU，并且在模型初始化阶段通常不涉及到具体的计算，因此可以在 CPU 上 阅读全文
posted @ 2023-04-05 15:04 CodeBunny 阅读(28) 评论(0) 推荐(0) 编辑