BaseMethod

add_model_specific_args

learnable_params

configure_optimizers

forward

multicrop_forward

_base_shared_step

training_step

validation_step

on_validation_epoch_end

BaseMomentumMethod

learnable_params

momentum_pairs

add_model_specific_args

on_train_start

momentum_forward

_shared_step_momentum

training_step

on_train_batch_end

validation_step

on_validation_epoch_end