↓ 3 callersMethod__init__(self, gate_channels, reduction_ratio=16, pool_types=('avg', 'max'), no_spatial=False)
code/models/modules/cbam_fromdehazeflow.py:90
↓ 2 callersMethodF(self, in_channels, out_channels, hidden_channels, kernel_hidden=1, n_hidden_layers=1)
code/models/modules/FlowAffineCouplingsAblation.py:170
↓ 2 callersMethod__init__(self, in_channels, out_channels,
kernel_size=[3, 3], stride=[1, 1],
padding
code/models/modules/flow.py:34
↓ 1 callersMethod__init__(self, optimizer, milestones, restarts=None, weights=None, gamma=0.1,
clear_state=False, last
code/models/lr_scheduler.py:9
↓ 1 callersMethodarch_FlowStep(self, H, K, LU_decomposed, W, actnorm_scale, affineInCh, flow_coupling, flow_permutation,
code/models/modules/FlowUpsamplerNet.py:118
↓ 1 callersMethodarch_additionalFlowAffine(self, H, LU_decomposed, W, actnorm_scale, hidden_channels, opt)
code/models/modules/FlowUpsamplerNet.py:165
↓ 1 callersMethodarch_split(self, H, W, L, levels, opt, opt_get)
code/models/modules/FlowUpsamplerNet.py:145
↓ 1 callersMethoddecode(self, rrdbResults, z, eps_std=None, epses=None, logdet=0.0, y_onehot=None,return_middleflow=False)
code/models/modules/FlowUpsamplerNet.py:263
↓ 1 callersMethodforward(self, input, logdet=None, reverse=False, offset_mask=None, logs_offset=None, bias_offset=None)
code/models/modules/FlowActNorms.py:81
↓ 1 callersMethodforward_split2d(self, epses, fl_fea, layer, logdet, reverse, rrdbResults, y_onehot=None)
code/models/modules/FlowUpsamplerNet.py:255
↓ 1 callersMethodforward_split2d_reverse(self, eps_std, epses, fl_fea, layer, rrdbResults, logdet, y_onehot=None)
code/models/modules/FlowUpsamplerNet.py:297
↓ 1 callersMethodreverse_flow(self, lr, z, y_onehot, eps_std, epses=None, lr_enc=None, add_gt_noise=True)
code/models/modules/LLFlow_arch.py:111