(self)
| 401 | teacher_param.data = momentum * teacher_param.data + (1 - momentum) * student_param.data |
| 402 | |
| 403 | def get_ssl_params(self): |
| 404 | ssl_params = [] |
| 405 | if self.enable_teacher and self.dino_head is not None: |
| 406 | ssl_params.extend(list(self.dino_head.parameters())) |
| 407 | return ssl_params |
| 408 | |
| 409 | |
| 410 | def get_teacher_forward_outputs(self, global_crops, n_global_crops, mask_indices_list, |
nothing calls this directly
no outgoing calls
no test coverage detected