Method__init__(self, optimizer, T_period, restarts=None, weights=None, eta_min=0, last_epoch=-1)
code/models/lr_scheduler.py:38
Method__init__(self, device='0', vgg=None, weights=None, indices=None, normalize=True)
code/models/modules/loss.py:51
Method__init__(self, device='0', vgg=None, weights=None, indices=None, normalize=True)
code/models/modules/loss.py:78
Method__init__(self, in_channels, out_channels,
kernel_size=[3, 3], stride=[1, 1],
padding
code/models/modules/flow.py:56
Method__init__(self, in_planes, out_planes, kernel_size, stride=1, padding=0, dilation=1, groups=1, relu=True, bn=True, bias
code/models/modules/cbam_fromdehazeflow.py:7
Method__init__(self, in_nc, out_nc, nf, nb, gc=32, scale=4, opt=None)
code/models/modules/RRDBNet_arch.py:52
Method__init__(self, in_nc, out_nc, nf, nb, gc=32, scale=4, opt=None)
code/models/modules/ConditionEncoder.py:63
Method__init__(self, in_nc, out_nc, nf, nb, gc=32, scale=4, opt=None)
code/models/modules/ConditionEncoder.py:154
Method__init__(self, in_channels, hidden_channels,
actnorm_scale=1.0, flow_permutation="invconv", flow_coup
code/models/modules/FlowStep.py:33
Method__init__(self, in_size, out_size, downsample, relu_slope, use_csff=False)
code/models/modules/FlowAffineCouplingsAblation.py:236
Method__init__(self, image_shape, hidden_channels, K, L=None,
actnorm_scale=1.0,
flow_perm
code/models/modules/FlowUpsamplerNet.py:17
Method__init__(self, num_channels, logs_eps=0, cond_channels=0, position=None, consume_ratio=0.5, opt=None)
code/models/modules/Split.py:14
Method__init__(self, in_nc, out_nc, nf, nb, gc=32, scale=4, K=None, opt=None, step=None)
code/models/modules/LLFlow_arch.py:22
Methodencode(self, gt, rrdbResults, logdet=0.0, epses=None, y_onehot=None)
code/models/modules/FlowUpsamplerNet.py:207
Functionflow_warpWarp an image or feature map with optical flow Args: x (Tensor): size (N, C, H, W) flow (Tensor): size (N, H, W, 2), normal value
code/models/modules/module_util.py:58