Module.parameters（）如何找到参数？

作者: 哦哦了
发布时间: 2024-04-25 03:38:00 (2月前)
转自：

2 条回复

0#
回复此人
那年 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 这很简单，只需通过元编程查看属性并检查其类型 </p> <pre> <code> class Example(): def __init__(self): self.special_thing = nn.Parameter(torch.rand(2)) self.something_else = "ok" def get_parameters(self): for key, value in self.__dict__.items(): if type(value) == nn.Parameter: print(key, "is a parameter!") e = Example() e.get_parameters() # => special_thing is a parameter! </code> </pre> </DIV>

编辑

登录后才能参与评论