Get all parameters from individual torch.nn.Module
(self)
| 94 | self.var_names_to_load.add(name) |
| 95 | |
| 96 | def parameters(self): |
| 97 | """Get all parameters from individual torch.nn.Module""" |
| 98 | parameters = [] |
| 99 | |
| 100 | # base model |
| 101 | isModel = lambda net: isinstance(net, BaseModel) |
| 102 | nets = inspect.getmembers(self, isModel) # name, net |
| 103 | for name, net in nets: |
| 104 | parameters.append(net.parameters()) |
| 105 | |
| 106 | # nn.module |
| 107 | isNNModule = lambda net: isinstance(net, torch.nn.Module) |
| 108 | nets = inspect.getmembers(self, isNNModule) # name, net |
| 109 | for name, net in nets: |
| 110 | parameters.append(net.parameters()) |
| 111 | |
| 112 | parameters = itertools.chain(*parameters) |
| 113 | return parameters |
| 114 | |
| 115 | def to(self, device: torch.device): |
| 116 | """ |
no outgoing calls