SimSiamHook

class mmpretrain.engine.hooks.SimSiamHook(fix_pred_lr, lr, adjust_by_epoch=True)

Hook for SimSiam.

This hook is for SimSiam to fix learning rate of predictor.

Parameters:

• fix_pred_lr (bool) – whether to fix the lr of predictor or not.

• lr (float) – the value of fixed lr.

• adjust_by_epoch (bool, optional) – whether to set lr by epoch or iter. Defaults to True.

before_train_epoch(runner)

fix lr of predictor by epoch.

before_train_iter(runner, batch_idx, data_batch=None)

fix lr of predictor by iter.