DenseCLHook

class mmpretrain.engine.hooks.DenseCLHook(start_iters=1000)

Hook for DenseCL.

This hook includes loss_lambda warmup in DenseCL. Borrowed from the authors’ code: https://github.com/WXinlong/DenseCL.

Parameters:

start_iters (int) – The number of warmup iterations to set loss_lambda=0. Defaults to 1000.

before_train(runner)

Obtain loss_lambda from algorithm.

before_train_iter(runner, batch_idx, data_batch=None)

Adjust loss_lambda every train iter.