basecls.engine.hooks#
- class basecls.engine.hooks.CheckpointHook(save_dir=None, save_every_n_epoch=1)[源代码]#
基类:
BaseHook
Hook for managing checkpoints during training.
Effect during
after_epoch
andafter_train
procedure.- 参数
- class basecls.engine.hooks.EvalHook(save_dir=None, eval_every_n_epoch=1)[源代码]#
基类:
BaseHook
Hook for evaluating during training.
Effect during
after_epoch
andafter_train
procedure.- 参数
- class basecls.engine.hooks.LoggerHook(log_every_n_iter=20)[源代码]#
基类:
BaseHook
Hook for logging during training.
Effect during
before_train
,after_train
,before_iter
andafter_iter
procedure.- 参数
log_every_n_iter (
int
) – interval for logging. Default:20
- class basecls.engine.hooks.LRSchedulerHook[源代码]#
基类:
BaseHook
Hook for learning rate scheduling during training.
Effect during
before_epoch
procedure.- get_lr_factor(cfg, epoch_id)[源代码]#
Calculate learning rate factor.
It supports
"step"
,"linear"
,"cosine"
,"exp"
, and"rel_exp"
schedule.
- total_lr#
Total learning rate.
- class basecls.engine.hooks.PreciseBNHook(precise_every_n_epoch=1)[源代码]#
基类:
BaseHook
Hook for precising BN during training.
Effect during
after_epoch
procedure.- 参数
precise_every_n_epoch (
int
) – interval for precising BN. Default:1
- class basecls.engine.hooks.ResumeHook(save_dir=None, resume=False)[源代码]#
基类:
BaseHook
Hook for resuming training process.
Effect during
before_train
procedure.- 参数