SwAV

add_model_specific_args

learnable_params

on_train_start

forward

get_assignments

training_step

on_after_backward