Method__init__(self, *, image_size, patch_size, num_classes, dim, transformer, channels = 3)
vit_pytorch/efficient.py:6
Method__init__(self, dim, depth, heads, mlp_dim, attn_dropout, ff_dropout)
vit_pytorch/vit_pytorch.py:66
Method__init__(self, *, image_size, patch_size, num_classes, dim, depth, heads, mlp_dim, channels = 3, attn_dropout = 0., ff
vit_pytorch/vit_pytorch.py:118
Method__init__(self, factors=[1, 2, 4, 8, 16, 32])
datasets/augmentations.py:32
Method__init__(self, source_root, target_root, opts, target_transform=None, source_transform=None)
datasets/images_dataset.py:10
Method__init__(self, input_size, num_layers, mode='ir', drop_ratio=0.4, affine=True)
models/encoders/model_irse.py:10
Method__init__(
self, in_channel, out_channel, kernel_size, stride=1, padding=0, bias=True
)
models/stylegan2/model.py:91
Method__init__(
self, in_dim, out_dim, bias=True, bias_init=0, lr_mul=1, activation=None
)
models/stylegan2/model.py:129
Method__init__(self, in_channel, style_dim, upsample=True, blur_kernel=[1, 3, 3, 1])
models/stylegan2/model.py:341
Method__init__(self, in_channel, out_channel, blur_kernel=[1, 3, 3, 1])
models/stylegan2/model.py:600
Method__init__(self, size, channel_multiplier=2, blur_kernel=[1, 3, 3, 1])
models/stylegan2/model.py:621