(self, meta_weights, inner_weights, lr_meta)
| 144 | self.optimizer_inner.load_state_dict(self.optimizer_inner_state) |
| 145 | |
| 146 | def fish(self, meta_weights, inner_weights, lr_meta): |
| 147 | meta_weights = ParamDict(meta_weights) |
| 148 | inner_weights = ParamDict(inner_weights) |
| 149 | meta_weights += lr_meta * (inner_weights - meta_weights) |
| 150 | return meta_weights |
| 151 | |
| 152 | def update(self, minibatches, unlabeled=None): |
| 153 | self.create_clone(minibatches[0][0].device) |