(self)
| 135 | return self.ema_model.eval() |
| 136 | |
| 137 | def restore_ema_model_device(self): |
| 138 | device = self.initted.device |
| 139 | self.ema_model.to(device) |
| 140 | |
| 141 | def get_params_iter(self, model): |
| 142 | for name, param in model.named_parameters(): |
nothing calls this directly
no outgoing calls
no test coverage detected