↓ 11 callersMethod__init__(self, in_features, hidden_features=None, out_features=None, act_layer=gelu, drop=0.)
models_search/ViT_custom_scale2.py:77
↓ 11 callersMethod__init__(self, in_features, hidden_features=None, out_features=None, act_layer=gelu, drop=0.)
models_search/Celeba256_dis.py:79
↓ 11 callersMethod__init__(self, in_features, hidden_features=None, out_features=None, act_layer=gelu, drop=0.)
models_search/ViT_scale3_local_new_rp.py:77
↓ 11 callersMethod__init__(self, in_features, hidden_features=None, out_features=None, act_layer=gelu, drop=0.)
models_search/ViT_custom_scale2_rp_noise.py:77
↓ 11 callersMethod__init__(self, in_features, hidden_features=None, out_features=None, act_layer=gelu, drop=0.)
models_search/Celeba256_gen.py:51
↓ 10 callersMethod__init__(self, in_features, hidden_features=None, out_features=None, act_layer=gelu, drop=0.)
models_search/ViT_custom_rp.py:53
↓ 10 callersMethod__init__(self, in_features, hidden_features=None, out_features=None, act_layer=gelu, drop=0.)
models_search/ViT_custom_local544444_256_rp_noise.py:48
↓ 10 callersMethod__init__(self, in_features, hidden_features=None, out_features=None, act_layer=gelu, drop=0.)
models_search/ViT_custom_local544444_256_rp.py:48
↓ 10 callersMethod__init__(self, in_features, hidden_features=None, out_features=None, act_layer=gelu, drop=0.)
models_search/ViT_custom.py:53
↓ 4 callersFunction_conv2d_gradfix(transpose, weight_shape, stride, padding, output_padding, dilation, groups)
torch_utils/ops/conv2d_gradfix.py:68
↓ 4 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_custom_local544444_256_rp_noise.py:210
↓ 4 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_custom_local544444_256_rp.py:203
↓ 2 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/Celeba256_dis.py:185
↓ 2 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/Celeba256_gen.py:281