pytorch model.to(设备)返回什么?它的文档在哪里?

eqqqjvef  于 2023-05-17  发布在  其他
关注(0)|答案(1)|浏览(129)

我找到了models的文档,但没有找到www.example.com()的特定API文档model.to。
ChatGPT告诉了我相互矛盾的信息。它返回模型的副本,并且它返回模型的引用。
我试着理解这段代码:

modelBigram0 = BigramLanguageModel(vocab_size)
modelBigram1 = modelBigram0.to(device)

如果我有model.to()的文档,这将有所帮助。

f0brbegy

f0brbegy1#

方法对模块进行就地修改。它用于在设备(cpu/gpu)之间移动模型。返回值是对调用对象的自引用。

文档在这里,正如@Carcigenicate已经在评论中指出的那样。
根据文档的示例用法:

modelBigram0 = BigramLanguageModel(vocab_size)
modelBigram0.to(device)

不需要使用两个独立的变量。在你的例子中,它们都包含对同一对象的引用,该对象只在系统内存或GPU内存中存储一次。

相关问题