Method
__init__
(self,
module,
mean_param,
num_stages=1,
append_params=True,
learn_mean=False,
detach_mean=False,
dim=1,
**kwargs)
Source from the content-addressed store, hash-verified
| 16 | class IterativeRegression(nn.Module): |
| 17 | """Regressor for ExPose Head.""" |
| 18 | def __init__(self, |
| 19 | module, |
| 20 | mean_param, |
| 21 | num_stages=1, |
| 22 | append_params=True, |
| 23 | learn_mean=False, |
| 24 | detach_mean=False, |
| 25 | dim=1, |
| 26 | **kwargs): |
| 27 | super(IterativeRegression, self).__init__() |
| 28 | self.module = module |
| 29 | self._num_stages = num_stages |
| 30 | self.dim = dim |
| 31 | |
| 32 | if learn_mean: |
| 33 | self.register_parameter( |
| 34 | 'mean_param', nn.Parameter(mean_param, requires_grad=True)) |
| 35 | else: |
| 36 | self.register_buffer('mean_param', mean_param) |
| 37 | |
| 38 | self.append_params = append_params |
| 39 | self.detach_mean = detach_mean |
| 40 | |
| 41 | def get_mean(self): |
| 42 | """Get the initial mean param.""" |
Callers
nothing calls this directly
Tested by
no test coverage detected