optimizer_cfg = copy.deepcopy(cfg) constructor_type = optimizer_cfg.pop('constructor','DefaultOptimizerConstructor') paramwise_cfg = optimizer_cfg.pop('paramwise_cfg',None) optim_constructor = build_optimizer_constructor(dict(type=constructor_type, optimizer_cfg=optimizer_cfg, paramwise_cfg=paramwi...
(cfg): return build_from_cfg(cfg, OPTIMIZER_BUILDERS) # 创建优化器 def build_optimizer(model, cfg): optimizer_cfg = copy.deepcopy(cfg) constructor_type = optimizer_cfg.pop('constructor', 'DefaultOptimizerConstructor') paramwise_cfg = optimizer_cfg.pop('paramwise_cfg', None) optim_...
optimizer = dict(_delete_=True, type='AdamW', lr=0.0001, betas=(0.9, 0.999), weight_decay=0.05, paramwise_cfg=dict(custom_keys={'absolute_pos_embed': dict(decay_mult=0.), 'relative_position_bias_table': dict(decay_mult=0.), 'norm': dict(decay_mult=0.)})) lr_config = dict(s...
I am getting following error for this command in Google Colab: !python tools/train.py configs/swin/upernet_swin_tiny_patch4_window7_512x512_160k_mfdataset.py --cfg-options model.pretrained = checkpoints/swin_tiny_patch4_window7_224.pth --gpus=1 --gpu-ids=1 No CUDA runtime is found, ...
constructor_type = optimizer_cfg.pop('constructor','DefaultOptimizerConstructor') paramwise_cfg = optimizer_cfg.pop('paramwise_cfg',None) optim_constructor = build_optimizer_constructor(dict(type=constructor_type, optimizer_cfg=optimizer_cfg,