# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved. """Configs.""" from fvcore.common.config import CfgNode from . import custom_config # ----------------------------------------------------------------------------- # Config definition # ----------------------------------------------------------------------------- _C = CfgNode() # ---------------------------------------------------------------------------- # # Batch norm options # ---------------------------------------------------------------------------- # _C.BN = CfgNode() # Precise BN stats. _C.BN.USE_PRECISE_STATS = False # Number of samples use to compute precise bn. _C.BN.NUM_BATCHES_PRECISE = 200 # Weight decay value that applies on BN. _C.BN.WEIGHT_DECAY = 0.0 # Norm type, options include `batchnorm`, `sub_batchnorm`, `sync_batchnorm` _C.BN.NORM_TYPE = "batchnorm" # Parameter for SubBatchNorm, where it splits the batch dimension into # NUM_SPLITS splits, and run BN on each of them separately independently.
"""Configs.""" from fvcore.common.config import CfgNode from . import custom_config new_allowed = True # ----------------------------------------------------------------------------- # Config definition # ----------------------------------------------------------------------------- _C = CfgNode(new_allowed=new_allowed) # ---------------------------------------------------------------------------- # # Batch norm options # ---------------------------------------------------------------------------- # _C.BN = CfgNode(new_allowed=new_allowed) # Precise BN stats. _C.BN.USE_PRECISE_STATS = False # Number of samples use to compute precise bn. _C.BN.NUM_BATCHES_PRECISE = 200 # Weight decay value that applies on BN. _C.BN.WEIGHT_DECAY = 0.0 # Norm type, options include `batchnorm`, `sub_batchnorm`, `sync_batchnorm` _C.BN.NORM_TYPE = "batchnorm" # Parameter for SubBatchNorm, where it splits the batch dimension into # NUM_SPLITS splits, and run BN on each of them separately independently.