Shortcuts

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.

参数:

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.

Read the Docs v: latest
Versions
latest
stable
mmcls-1.x
mmcls-0.x
dev
Downloads
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.