↓ 6 callersMethod__init__(self, args, is_cropped = False, root = '', dstype = 'clean', replicates = 1)
models/flownet2_pytorch/datasets.py:31
↓ 6 callersFunctioni_conv(batchNorm, in_planes, out_planes, kernel_size=3, stride=1, bias = True)
models/flownet2_pytorch/networks/submodules.py:20
↓ 4 callersFunctionget_skipped_frames_sparse(B_all, B, t_scales, tD, n_frames_load, i, is_flow=False)
models/vid2vid_model_D.py:304
↓ 2 callersFunctionsave_models(opt, epoch, epoch_iter, total_steps, visualizer, iter_path, modelG, modelD, end_of_epoch=False)
models/models.py:135
↓ 2 callersFunctiontrain(args, epoch, start_iteration, data_loader, model, optimizer, logger, is_validate=False, offset=0)
models/flownet2_pytorch/main.py:244
↓ 1 callersMethod__init__(self, pad_size=0, kernel_size=0, max_displacement=0, stride1=1, stride2=2, corr_multiply=1)
models/flownet2_pytorch/networks/correlation_package/correlation.py:48