↓ 1 callersMethodaccumulate_gradients(self, phase, real_img, real_c, gen_z, gen_c, gain, cur_nimg)
draggan/stylegan2/training/loss.py:20
↓ 1 callersFunctiondrag_gan(
g_ema,
latent: torch.Tensor,
noise,
F,
handle_points,
target_points,
mask,
m
draggan/deprecated/api.py:103
↓ 1 callersFunctionmodulated_conv2d(
x, # Input tensor: [batch_size, in_channels, in_height, in_width]
w,
draggan/stylegan2/training/networks_stylegan3.py:26
Method__init__(
self, in_channel, out_channel, kernel_size, stride=1, padding=0, bias=True
)
draggan/deprecated/stylegan2/model.py:110
Method__init__(
self, in_dim, out_dim, bias=True, bias_init=0, lr_mul=1, activation=None
)
draggan/deprecated/stylegan2/model.py:148
Method__init__(
self,
in_channel,
out_channel,
kernel_size,
style_dim,
demod
draggan/deprecated/stylegan2/model.py:185
Method__init__(
self,
in_channel,
out_channel,
kernel_size,
style_dim,
upsam
draggan/deprecated/stylegan2/model.py:348
Method__init__(self, in_channel, style_dim, upsample=True, blur_kernel=[1, 3, 3, 1])
draggan/deprecated/stylegan2/model.py:385
Method__init__(
self,
size,
style_dim,
n_mlp,
channel_multiplier=2,
blur_ker
draggan/deprecated/stylegan2/model.py:407
Method__init__(
self,
in_channel,
out_channel,
kernel_size,
downsample=False,
draggan/deprecated/stylegan2/model.py:590
Method__init__(self, in_channel, out_channel, blur_kernel=[1, 3, 3, 1])
draggan/deprecated/stylegan2/model.py:635
Method__init__(self, size, channel_multiplier=2, blur_kernel=[1, 3, 3, 1])
draggan/deprecated/stylegan2/model.py:656
Method__init__(self, model='net-lin', net='alex', colorspace='rgb', spatial=False, use_gpu=True, gpu_ids=[0])
draggan/deprecated/stylegan2/lpips/util.py:14
Method__init__(self, dataset, rank=0, num_replicas=1, shuffle=True, seed=0, window_size=0.5)
draggan/stylegan2/torch_utils/misc.py:110
Method__init__(self, device, G, D, augment_pipe=None, r1_gamma=10, style_mixing_prob=0, pl_weight=0, pl_batch_shrink=2, pl_d
draggan/stylegan2/training/loss.py:26