Table of Contents
Shortcuts

ClassNumCheckHook

class mmpretrain.engine.hooks.ClassNumCheckHook[source]

Class Number Check HOOK.

before_test(runner)[source]

Check whether the test dataset is compatible with head.

Parameters:

(obj (runner) – IterBasedRunner): Iter based Runner.

before_train(runner)[source]

Check whether the training dataset is compatible with head.

Parameters:

(obj (runner) – IterBasedRunner): Iter based Runner.

before_val(runner)[source]

Check whether the validation dataset is compatible with head.

Parameters:

(obj (runner) – IterBasedRunner): Iter based Runner.