(
self,
global_crops,
n_global_crops,
mask_indices_list,
n_masked_patches,
upperbound,
local_crops,
masks
)
| 363 | return self.get_reconstruction_outputs(reconstruction_image) |
| 364 | |
| 365 | def forward_ssl_learning( |
| 366 | self, |
| 367 | global_crops, |
| 368 | n_global_crops, |
| 369 | mask_indices_list, |
| 370 | n_masked_patches, |
| 371 | upperbound, |
| 372 | local_crops, |
| 373 | masks |
| 374 | ): |
| 375 | teacher_outputs = self.get_teacher_forward_outputs( |
| 376 | global_crops, n_global_crops, mask_indices_list, |
| 377 | n_masked_patches, upperbound |
| 378 | ) |
| 379 | |
| 380 | student_outputs = self.get_student_ssl_outputs( |
| 381 | global_crops, local_crops, masks, mask_indices_list, |
| 382 | n_masked_patches, upperbound |
| 383 | ) |
| 384 | |
| 385 | return teacher_outputs, student_outputs |
| 386 | |
| 387 | |
| 388 | def update_teacher(self, momentum: float): |
no test coverage detected