NNCLR

add_model_specific_args

learnable_params

dequeue_and_enqueue

find_nn

forward

training_step