↓ 1 callersFunctionbchw_to_bhwcPermutes a tensor from the shape (B, C, H, W) to (B, H, W, C).
src/train_acdc/lib/models_timm/swin_transformer_v2_cr.py:104
↓ 1 callersFunctioncosine_scheduler(base_value, final_value, epochs, niter_per_ep, warmup_epochs=0, start_warmup_value=0)
src/train_synase/trainer_unet.py:88
↓ 1 callersFunctioncosine_scheduler(base_value, final_value, epochs, niter_per_ep, warmup_epochs=0, start_warmup_value=0)
src/train_synase/trainer.py:96
↓ 1 callersFunctioncreate_byob_stages(
cfg: ByoModelCfg, drop_path_rate: float, output_stride: int, stem_feat: Dict[str, Any],
feat
src/train_acdc/lib/models_timm/byobnet.py:1409
↓ 1 callersFunctioncreate_byob_stem(in_chs, out_chs, stem_type='', pool_type='', feat_prefix='stem', layers: LayerFn = None)
src/train_acdc/lib/models_timm/byobnet.py:1322
↓ 1 callersFunctioncreate_csp_stem(
in_chans=3,
out_chs=32,
kernel_size=3,
stride=2,
pool='',
pa
src/train_acdc/lib/models_timm/cspnet.py:711
↓ 1 callersFunctioncreate_stem(in_chs, out_chs, stem_type='', conv_layer=None, act_layer=None, preact_feature=True)
src/train_acdc/lib/models_timm/nfnet.py:370
↓ 1 callersFunctiondownsample_avg(
in_channels, out_channels, kernel_size, stride=1, dilation=1, first_dilation=None, norm_layer=None)
src/train_acdc/lib/models_timm/resnet.py:484
↓ 1 callersFunctiondownsample_conv(
in_channels, out_channels, kernel_size, stride=1, dilation=1, first_dilation=None, norm_layer=None)
src/train_acdc/lib/models_timm/resnet.py:470
↓ 1 callersFunctiondownsample_conv(in_chs, out_chs, kernel_size=1, stride=1, dilation=1, norm_layer=None, preact=False)
src/train_acdc/lib/models_timm/regnet.py:202