02 2024 档案

【极简】Pytorch中的register_buffer()
摘要：register buffer 定义模型能用torch.save保存的、但是不更新参数。 使用：只要是nn.Module的子类就能直接self.调用使用： class A(nn.Module): #... self.register_buffer( 'betas', torch.linspace(b 阅读全文
posted @ 2024-02-01 10:22 ZephyrYin 阅读(374) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示