摘要:
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
阅读(606)
评论(0)
推荐(0)
浙公网安备 33010602011771号