↓ 5 callersFunctionget_1x1(in_dim, out_dim, zero_bias=True, zero_weights=False, groups=1, scaled=False)
helpers/imle_helpers.py:35
↓ 4 callersMethodcalc_dists_existing(self, dataset_tensor, gen, dists=None, dists_lpips = None, dists_l2 = None, latents=None, to_update=None, sno
sampler.py:253
↓ 3 callersFunctiontraining_step_imle(H, n, targets, latents, snoise, imle, ema_imle, optimizer, loss_fn)
train.py:33
↓ 2 callersFunctiongenerate_images_initial(H, sampler, orig, initial, snoise, shape, imle, ema_imle, fname, logprint, experiment=None)
visual/utils.py:42
↓ 2 callersFunctionget_3x3(in_dim, out_dim, zero_bias=True, zero_weights=False, groups=1, scaled=False)
helpers/imle_helpers.py:31
↓ 2 callersFunctionget_conv(in_dim, out_dim, kernel_size, stride, padding, zero_bias=True, zero_weights=False, groups=1, scaled=False)
helpers/imle_helpers.py:22