Safemotion Lib
Loading...
Searching...
No Matches
Functions | Variables
fastreid.modeling.backbones.regnet.config Namespace Reference

Functions

 assert_and_infer_cfg (cache_urls=True)
 
 cache_cfg_urls ()
 
 dump_cfg ()
 
 load_cfg (out_dir, cfg_dest="config.yaml")
 
 load_cfg_fom_args (description="Config file options.")
 

Variables

 _C = CfgNode()
 
 cfg = _C
 
 MODEL
 
 TYPE
 
 DEPTH
 
 NUM_CLASSES
 
 LOSS_FUN
 
 RESNET
 
 TRANS_FUN
 
 NUM_GROUPS
 
 WIDTH_PER_GROUP
 
 STRIDE_1X1
 
 ANYNET
 
 STEM_TYPE
 
 STEM_W
 
 BLOCK_TYPE
 
 DEPTHS
 
 WIDTHS
 
 STRIDES
 
 BOT_MULS
 
 GROUP_WS
 
 SE_ON
 
 SE_R
 
 REGNET
 
 STRIDE
 
 W0
 
 WA
 
 WM
 
 GROUP_W
 
 BOT_MUL
 
 EN
 
 EXP_RATIOS
 
 KERNELS
 
 HEAD_W
 
 DC_RATIO
 
 DROPOUT_RATIO
 
 BN
 
 EPS
 
 MOM
 
 USE_PRECISE_STATS
 
 NUM_SAMPLES_PRECISE
 
 ZERO_INIT_FINAL_GAMMA
 
 USE_CUSTOM_WEIGHT_DECAY
 
 CUSTOM_WEIGHT_DECAY
 
 OPTIM
 
 BASE_LR
 
 LR_POLICY
 
 GAMMA
 
 STEPS
 
 LR_MULT
 
 MAX_EPOCH
 
 MOMENTUM
 
 DAMPENING
 
 NESTEROV
 
 WEIGHT_DECAY
 
 WARMUP_FACTOR
 
 WARMUP_EPOCHS
 
 TRAIN
 
 DATASET
 
 SPLIT
 
 BATCH_SIZE
 
 IM_SIZE
 
 EVAL_PERIOD
 
 CHECKPOINT_PERIOD
 
 AUTO_RESUME
 
 WEIGHTS
 
 TEST
 
 DATA_LOADER
 
 NUM_WORKERS
 
 PIN_MEMORY
 
 MEM
 
 RELU_INPLACE
 
 CUDNN
 
 BENCHMARK
 
 PREC_TIME
 
 WARMUP_ITER
 
 NUM_ITER
 
 NUM_GPUS
 
 OUT_DIR
 
 CFG_DEST
 
 RNG_SEED
 
 LOG_DEST
 
 LOG_PERIOD
 
 DIST_BACKEND
 
 HOST
 
 PORT_RANGE
 
 DOWNLOAD_CACHE
 

Detailed Description

Configuration file (powered by YACS).

Function Documentation

◆ assert_and_infer_cfg()

fastreid.modeling.backbones.regnet.config.assert_and_infer_cfg ( cache_urls = True)
Checks config values invariants.

Definition at line 372 of file config.py.

372def assert_and_infer_cfg(cache_urls=True):
373 """Checks config values invariants."""
374 err_str = "The first lr step must start at 0"
375 assert not _C.OPTIM.STEPS or _C.OPTIM.STEPS[0] == 0, err_str
376 data_splits = ["train", "val", "test"]
377 err_str = "Data split '{}' not supported"
378 assert _C.TRAIN.SPLIT in data_splits, err_str.format(_C.TRAIN.SPLIT)
379 assert _C.TEST.SPLIT in data_splits, err_str.format(_C.TEST.SPLIT)
380 err_str = "Mini-batch size should be a multiple of NUM_GPUS."
381 assert _C.TRAIN.BATCH_SIZE % _C.NUM_GPUS == 0, err_str
382 assert _C.TEST.BATCH_SIZE % _C.NUM_GPUS == 0, err_str
383 err_str = "Log destination '{}' not supported"
384 assert _C.LOG_DEST in ["stdout", "file"], err_str.format(_C.LOG_DEST)
385 if cache_urls:
386 cache_cfg_urls()
387
388

◆ cache_cfg_urls()

fastreid.modeling.backbones.regnet.config.cache_cfg_urls ( )
Download URLs in config, cache them, and rewrite cfg to use cached file.

Definition at line 389 of file config.py.

389def cache_cfg_urls():
390 """Download URLs in config, cache them, and rewrite cfg to use cached file."""
391 _C.TRAIN.WEIGHTS = cache_url(_C.TRAIN.WEIGHTS, _C.DOWNLOAD_CACHE)
392 _C.TEST.WEIGHTS = cache_url(_C.TEST.WEIGHTS, _C.DOWNLOAD_CACHE)
393
394

◆ dump_cfg()

fastreid.modeling.backbones.regnet.config.dump_cfg ( )
Dumps the config to the output directory.

Definition at line 395 of file config.py.

395def dump_cfg():
396 """Dumps the config to the output directory."""
397 cfg_file = os.path.join(_C.OUT_DIR, _C.CFG_DEST)
398 with open(cfg_file, "w") as f:
399 _C.dump(stream=f)
400
401

◆ load_cfg()

fastreid.modeling.backbones.regnet.config.load_cfg ( out_dir,
cfg_dest = "config.yaml" )
Loads config from specified output directory.

Definition at line 402 of file config.py.

402def load_cfg(out_dir, cfg_dest="config.yaml"):
403 """Loads config from specified output directory."""
404 cfg_file = os.path.join(out_dir, cfg_dest)
405 _C.merge_from_file(cfg_file)
406
407

◆ load_cfg_fom_args()

fastreid.modeling.backbones.regnet.config.load_cfg_fom_args ( description = "Config file options.")
Load config from command line arguments and set any specified options.

Definition at line 408 of file config.py.

408def load_cfg_fom_args(description="Config file options."):
409 """Load config from command line arguments and set any specified options."""
410 parser = argparse.ArgumentParser(description=description)
411 help_s = "Config file location"
412 parser.add_argument("--cfg", dest="cfg_file", help=help_s, required=True, type=str)
413 help_s = "See pycls/core/config.py for all options"
414 parser.add_argument("opts", help=help_s, default=None, nargs=argparse.REMAINDER)
415 if len(sys.argv) == 1:
416 parser.print_help()
417 sys.exit(1)
418 args = parser.parse_args()
419 _C.merge_from_file(args.cfg_file)
420 _C.merge_from_list(args.opts)

Variable Documentation

◆ _C

fastreid.modeling.backbones.regnet.config._C = CfgNode()
protected

Definition at line 18 of file config.py.

◆ ANYNET

fastreid.modeling.backbones.regnet.config.ANYNET

Definition at line 64 of file config.py.

◆ AUTO_RESUME

fastreid.modeling.backbones.regnet.config.AUTO_RESUME

Definition at line 260 of file config.py.

◆ BASE_LR

fastreid.modeling.backbones.regnet.config.BASE_LR

Definition at line 202 of file config.py.

◆ BATCH_SIZE

fastreid.modeling.backbones.regnet.config.BATCH_SIZE

Definition at line 248 of file config.py.

◆ BENCHMARK

fastreid.modeling.backbones.regnet.config.BENCHMARK

Definition at line 314 of file config.py.

◆ BLOCK_TYPE

fastreid.modeling.backbones.regnet.config.BLOCK_TYPE

Definition at line 73 of file config.py.

◆ BN

fastreid.modeling.backbones.regnet.config.BN

Definition at line 176 of file config.py.

◆ BOT_MUL

fastreid.modeling.backbones.regnet.config.BOT_MUL

Definition at line 134 of file config.py.

◆ BOT_MULS

fastreid.modeling.backbones.regnet.config.BOT_MULS

Definition at line 85 of file config.py.

◆ cfg

fastreid.modeling.backbones.regnet.config.cfg = _C

Definition at line 22 of file config.py.

◆ CFG_DEST

fastreid.modeling.backbones.regnet.config.CFG_DEST

Definition at line 340 of file config.py.

◆ CHECKPOINT_PERIOD

fastreid.modeling.backbones.regnet.config.CHECKPOINT_PERIOD

Definition at line 257 of file config.py.

◆ CUDNN

fastreid.modeling.backbones.regnet.config.CUDNN

Definition at line 309 of file config.py.

◆ CUSTOM_WEIGHT_DECAY

fastreid.modeling.backbones.regnet.config.CUSTOM_WEIGHT_DECAY

Definition at line 193 of file config.py.

◆ DAMPENING

fastreid.modeling.backbones.regnet.config.DAMPENING

Definition at line 223 of file config.py.

◆ DATA_LOADER

fastreid.modeling.backbones.regnet.config.DATA_LOADER

Definition at line 288 of file config.py.

◆ DATASET

fastreid.modeling.backbones.regnet.config.DATASET

Definition at line 244 of file config.py.

◆ DC_RATIO

fastreid.modeling.backbones.regnet.config.DC_RATIO

Definition at line 167 of file config.py.

◆ DEPTH

fastreid.modeling.backbones.regnet.config.DEPTH

Definition at line 34 of file config.py.

◆ DEPTHS

fastreid.modeling.backbones.regnet.config.DEPTHS

Definition at line 76 of file config.py.

◆ DIST_BACKEND

fastreid.modeling.backbones.regnet.config.DIST_BACKEND

Definition at line 353 of file config.py.

◆ DOWNLOAD_CACHE

fastreid.modeling.backbones.regnet.config.DOWNLOAD_CACHE

Definition at line 360 of file config.py.

◆ DROPOUT_RATIO

fastreid.modeling.backbones.regnet.config.DROPOUT_RATIO

Definition at line 170 of file config.py.

◆ EN

fastreid.modeling.backbones.regnet.config.EN

Definition at line 140 of file config.py.

◆ EPS

fastreid.modeling.backbones.regnet.config.EPS

Definition at line 179 of file config.py.

◆ EVAL_PERIOD

fastreid.modeling.backbones.regnet.config.EVAL_PERIOD

Definition at line 254 of file config.py.

◆ EXP_RATIOS

fastreid.modeling.backbones.regnet.config.EXP_RATIOS

Definition at line 152 of file config.py.

◆ GAMMA

fastreid.modeling.backbones.regnet.config.GAMMA

Definition at line 208 of file config.py.

◆ GROUP_W

fastreid.modeling.backbones.regnet.config.GROUP_W

Definition at line 131 of file config.py.

◆ GROUP_WS

fastreid.modeling.backbones.regnet.config.GROUP_WS

Definition at line 88 of file config.py.

◆ HEAD_W

fastreid.modeling.backbones.regnet.config.HEAD_W

Definition at line 164 of file config.py.

◆ HOST

fastreid.modeling.backbones.regnet.config.HOST

Definition at line 356 of file config.py.

◆ IM_SIZE

fastreid.modeling.backbones.regnet.config.IM_SIZE

Definition at line 251 of file config.py.

◆ KERNELS

fastreid.modeling.backbones.regnet.config.KERNELS

Definition at line 161 of file config.py.

◆ LOG_DEST

fastreid.modeling.backbones.regnet.config.LOG_DEST

Definition at line 347 of file config.py.

◆ LOG_PERIOD

fastreid.modeling.backbones.regnet.config.LOG_PERIOD

Definition at line 350 of file config.py.

◆ LOSS_FUN

fastreid.modeling.backbones.regnet.config.LOSS_FUN

Definition at line 40 of file config.py.

◆ LR_MULT

fastreid.modeling.backbones.regnet.config.LR_MULT

Definition at line 214 of file config.py.

◆ LR_POLICY

fastreid.modeling.backbones.regnet.config.LR_POLICY

Definition at line 205 of file config.py.

◆ MAX_EPOCH

fastreid.modeling.backbones.regnet.config.MAX_EPOCH

Definition at line 217 of file config.py.

◆ MEM

fastreid.modeling.backbones.regnet.config.MEM

Definition at line 300 of file config.py.

◆ MODEL

fastreid.modeling.backbones.regnet.config.MODEL

Definition at line 28 of file config.py.

◆ MOM

fastreid.modeling.backbones.regnet.config.MOM

Definition at line 182 of file config.py.

◆ MOMENTUM

fastreid.modeling.backbones.regnet.config.MOMENTUM

Definition at line 220 of file config.py.

◆ NESTEROV

fastreid.modeling.backbones.regnet.config.NESTEROV

Definition at line 226 of file config.py.

◆ NUM_CLASSES

fastreid.modeling.backbones.regnet.config.NUM_CLASSES

Definition at line 37 of file config.py.

◆ NUM_GPUS

fastreid.modeling.backbones.regnet.config.NUM_GPUS

Definition at line 334 of file config.py.

◆ NUM_GROUPS

fastreid.modeling.backbones.regnet.config.NUM_GROUPS

Definition at line 52 of file config.py.

◆ NUM_ITER

fastreid.modeling.backbones.regnet.config.NUM_ITER

Definition at line 326 of file config.py.

◆ NUM_SAMPLES_PRECISE

fastreid.modeling.backbones.regnet.config.NUM_SAMPLES_PRECISE

Definition at line 186 of file config.py.

◆ NUM_WORKERS

fastreid.modeling.backbones.regnet.config.NUM_WORKERS

Definition at line 291 of file config.py.

◆ OPTIM

fastreid.modeling.backbones.regnet.config.OPTIM

Definition at line 199 of file config.py.

◆ OUT_DIR

fastreid.modeling.backbones.regnet.config.OUT_DIR

Definition at line 337 of file config.py.

◆ PIN_MEMORY

fastreid.modeling.backbones.regnet.config.PIN_MEMORY

Definition at line 294 of file config.py.

◆ PORT_RANGE

fastreid.modeling.backbones.regnet.config.PORT_RANGE

Definition at line 357 of file config.py.

◆ PREC_TIME

fastreid.modeling.backbones.regnet.config.PREC_TIME

Definition at line 320 of file config.py.

◆ REGNET

fastreid.modeling.backbones.regnet.config.REGNET

Definition at line 100 of file config.py.

◆ RELU_INPLACE

fastreid.modeling.backbones.regnet.config.RELU_INPLACE

Definition at line 303 of file config.py.

◆ RESNET

fastreid.modeling.backbones.regnet.config.RESNET

Definition at line 46 of file config.py.

◆ RNG_SEED

fastreid.modeling.backbones.regnet.config.RNG_SEED

Definition at line 344 of file config.py.

◆ SE_ON

fastreid.modeling.backbones.regnet.config.SE_ON

Definition at line 91 of file config.py.

◆ SE_R

fastreid.modeling.backbones.regnet.config.SE_R

Definition at line 94 of file config.py.

◆ SPLIT

fastreid.modeling.backbones.regnet.config.SPLIT

Definition at line 245 of file config.py.

◆ STEM_TYPE

fastreid.modeling.backbones.regnet.config.STEM_TYPE

Definition at line 67 of file config.py.

◆ STEM_W

fastreid.modeling.backbones.regnet.config.STEM_W

Definition at line 70 of file config.py.

◆ STEPS

fastreid.modeling.backbones.regnet.config.STEPS

Definition at line 211 of file config.py.

◆ STRIDE

fastreid.modeling.backbones.regnet.config.STRIDE

Definition at line 112 of file config.py.

◆ STRIDE_1X1

fastreid.modeling.backbones.regnet.config.STRIDE_1X1

Definition at line 58 of file config.py.

◆ STRIDES

fastreid.modeling.backbones.regnet.config.STRIDES

Definition at line 82 of file config.py.

◆ TEST

fastreid.modeling.backbones.regnet.config.TEST

Definition at line 269 of file config.py.

◆ TRAIN

fastreid.modeling.backbones.regnet.config.TRAIN

Definition at line 241 of file config.py.

◆ TRANS_FUN

fastreid.modeling.backbones.regnet.config.TRANS_FUN

Definition at line 49 of file config.py.

◆ TYPE

fastreid.modeling.backbones.regnet.config.TYPE

Definition at line 31 of file config.py.

◆ USE_CUSTOM_WEIGHT_DECAY

fastreid.modeling.backbones.regnet.config.USE_CUSTOM_WEIGHT_DECAY

Definition at line 192 of file config.py.

◆ USE_PRECISE_STATS

fastreid.modeling.backbones.regnet.config.USE_PRECISE_STATS

Definition at line 185 of file config.py.

◆ W0

fastreid.modeling.backbones.regnet.config.W0

Definition at line 122 of file config.py.

◆ WA

fastreid.modeling.backbones.regnet.config.WA

Definition at line 125 of file config.py.

◆ WARMUP_EPOCHS

fastreid.modeling.backbones.regnet.config.WARMUP_EPOCHS

Definition at line 235 of file config.py.

◆ WARMUP_FACTOR

fastreid.modeling.backbones.regnet.config.WARMUP_FACTOR

Definition at line 232 of file config.py.

◆ WARMUP_ITER

fastreid.modeling.backbones.regnet.config.WARMUP_ITER

Definition at line 323 of file config.py.

◆ WEIGHT_DECAY

fastreid.modeling.backbones.regnet.config.WEIGHT_DECAY

Definition at line 229 of file config.py.

◆ WEIGHTS

fastreid.modeling.backbones.regnet.config.WEIGHTS

Definition at line 263 of file config.py.

◆ WIDTH_PER_GROUP

fastreid.modeling.backbones.regnet.config.WIDTH_PER_GROUP

Definition at line 55 of file config.py.

◆ WIDTHS

fastreid.modeling.backbones.regnet.config.WIDTHS

Definition at line 79 of file config.py.

◆ WM

fastreid.modeling.backbones.regnet.config.WM

Definition at line 128 of file config.py.

◆ ZERO_INIT_FINAL_GAMMA

fastreid.modeling.backbones.regnet.config.ZERO_INIT_FINAL_GAMMA

Definition at line 189 of file config.py.