Method__init__(self, dataset, rank=0, num_replicas=1, shuffle=True, seed=0, window_size=0.5)
facelit/torch_utils/misc.py:114
Method__init__(self, G, G_kwargs, epsilon, space, sampling, crop, vgg16)
facelit/metrics/perceptual_path_length.py:38
Method__init__(self, G=None, G_kwargs={}, dataset_kwargs={}, num_gpus=1, rank=0, device=None, progress=None, cache=True)
facelit/metrics/metric_utils.py:26
Method__init__(self, tag=None, num_items=None, flush_interval=1000, verbose=False, progress_fn=None, pfn_lo=0, pfn_hi=1000,
facelit/metrics/metric_utils.py:154
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
facelit/training/loss.py:29
Method__init__(self, channels, img_resolution, sr_num_fp16_res, sr_antialias,
num_fp16_res=4, conv_clamp=Non
facelit/training/superresolution.py:30
Method__init__(self, channels, img_resolution, sr_num_fp16_res, sr_antialias,
num_fp16_res=4, conv_clamp=Non
facelit/training/superresolution.py:63
Method__init__(self, channels, img_resolution, sr_num_fp16_res, sr_antialias,
num_fp16_res=4, conv_clamp=Non
facelit/training/superresolution.py:95
Method__init__(self, channels, img_resolution, sr_num_fp16_res,
num_fp16_res=4, conv_clamp=None, channel_bas
facelit/training/superresolution.py:128
Method__init__(self, channels, img_resolution, sr_num_fp16_res, sr_antialias,
num_fp16_res=4, conv_clamp=Non
facelit/training/superresolution.py:265
Method__init__(self, in_channels, out_channels, w_dim, kernel_size=1, conv_clamp=None, channels_last=False)
facelit/training/networks_stylegan2.py:341
Method__init__(self,
xflip=0, rotate90=0, xint=0, xint_max=0.125,
scale=0, rotate=0, aniso=0, xfrac=0, scale
facelit/training/augment.py:125
Methodaccumulate_gradients(self, phase, real_img, real_c, gen_z, gen_c, gain, cur_nimg)
facelit/training/loss.py:91
Functioncompute_equivariance_metrics(opts, num_samples, batch_size, translate_max=0.125, rotate_max=1, compute_eqt_int=False, compute_eqt_frac=Fal
facelit/metrics/equivariance.py:194
Functioncompute_feature_stats_for_dataset(opts, detector_url, detector_kwargs, rel_lo=0, rel_hi=1, batch_size=64, data_loader_kwargs=None, max_items=No
facelit/metrics/metric_utils.py:198
Functioncompute_feature_stats_for_generator(opts, detector_url, detector_kwargs, rel_lo=0, rel_hi=1, batch_size=64, batch_gen=None, **stats_kwargs)
facelit/metrics/metric_utils.py:251
Functioncompute_ppl(opts, num_samples, epsilon, space, sampling, crop, batch_size)
facelit/metrics/perceptual_path_length.py:96
Functioncompute_pr(opts, max_real, num_gen, nhood_size, row_batch_size, col_batch_size)
facelit/metrics/precision_recall.py:38
Functionconv2d(input, weight, bias=None, stride=1, padding=0, dilation=1, groups=1)
facelit/torch_utils/ops/conv2d_gradfix.py:37
Functionconv_transpose2d(input, weight, bias=None, stride=1, padding=0, output_padding=0, groups=1, dilation=1)
facelit/torch_utils/ops/conv2d_gradfix.py:42