14 lines
401 B
Python
14 lines
401 B
Python
_base_ = 'resnet50_8xb32-coslr_in1k.py'
|
|
|
|
# Precise BN hook will update the bn stats, so this hook should be executed
|
|
# before CheckpointHook(priority of 'VERY_LOW') and
|
|
# EMAHook(priority of 'NORMAL') So set the priority of PreciseBNHook to
|
|
# 'ABOVENORMAL' here.
|
|
custom_hooks = [
|
|
dict(
|
|
type='PreciseBNHook',
|
|
num_samples=8192,
|
|
interval=1,
|
|
priority='ABOVE_NORMAL')
|
|
]
|