↓ 2 callersFunctionrandom_interp(H, sampler, shape, imle, fname, logprint, lat1=None, lat2=None, sn1=None, sn2=None)
visual/interpolate.py:13
↓ 2 callersFunctionrestore_params(model, path, local_rank, mpi_size, map_ddp=True, map_cpu=False, strict=True)
helpers/train_helpers.py:113
↓ 1 callersFunctionbacktrack(H, sampler, imle, preprocess_fn, data, logprint, training_step_imle)
helpers/imle_helpers.py:146
↓ 1 callersFunctiongenerate_rnd_nn(H, data, sampler, shape, imle, fname, logprint, preprocess_fn)
visual/generate_rnd_nn.py:5
↓ 1 callersFunctiongenerate_sample_nn(H, data, sampler, shape, ema_imle, fname, logprint, preprocess_fn)
visual/generate_sample_nn.py:6
↓ 1 callersFunctionnn_interp(H, data, sampler, shape, ema_imle, fname, logprint, preprocess_fn)
visual/nn_interplate.py:5
↓ 1 callersFunctionreconstruct(H, sampler, imle, preprocess_fn, images, latents, snoise, name, logprint, training_step_imle)
helpers/imle_helpers.py:203
↓ 1 callersFunctionspatial_vissual(H, sampler, shape, imle, fname, logprint, lat1=None, lat2=None, sn1=None, sn2=None)
visual/spatial_visual.py:6