↓ 1 callersFunctionget_fid(args, fid_stat, epoch, gen_net, num_img, gen_batch_size, val_batch_size, writer_dict=None, cls_idx=None)
utils/torch_fid_score.py:291
↓ 1 callersFunctiontrain(args, gen_net: nn.Module, dis_net: nn.Module, gen_optimizer, dis_optimizer, gen_avg_param, train_loader,
functions.py:70
↓ 1 callersFunctionwindow_partition Args: x: (B, H, W, C) window_size (int): window size Returns: windows: (num_windows*B, window_size, window_size, C)
models_search/ViT_scale3_local_new_rp.py:169
Method__init__(self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0., proj_drop=0., window_size=16)
models_search/ViT_custom_scale2.py:96
Method__init__(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0.,
dr
models_search/ViT_custom_scale2.py:201
Method__init__(self, depth, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0., drop_path=0.
models_search/ViT_custom_scale2.py:220
Method__init__(self, args, img_size=224, patch_size=16, in_chans=3, num_classes=10, embed_dim=384, depth=5,
models_search/ViT_custom_scale2.py:254
Method__init__(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0.,
dr
models_search/ViT_custom_scale2.py:390
Method__init__(self, args, img_size=32, patch_size=None, in_chans=3, num_classes=1, embed_dim=None, depth=7,
models_search/ViT_custom_scale2.py:410