ai-photo-gallery / configs /resnet /resnet50_8xb32-coslr-preciseBN_in1k.py
KyanChen's picture
init
f549064
raw
history blame
401 Bytes
_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')
]