↓ 7 callersMethod__init__(self, in_channel, style_dim, upsample=True, blur_kernel=[1, 3, 3, 1])
networks/prior_arch.py:288
↓ 1 callersMethod__init__(self, block=BasicBlock, layers=[3, 4, 6, 6, 3], strides=[2,1,2,1,2])
networks/w_encoder_arch.py:33
↓ 1 callersMethod_check_outliers_pad(self, crop_feature, start, end, max_lr_width, center_loc, extend_W)
networks/psp_encoder_arch.py:111
↓ 1 callersFunctionupfirdn2d_native(
input, kernel, up_x, up_y, down_x, down_y, pad_x0, pad_x1, pad_y0, pad_y1
)
op/upfirdn2d.py:168
Method__init__(self, block=BasicBlock, layers=[3, 4, 6, 6, 3], strides=[(2,2),(1,2),(2,2),(1,2),(2,2)])
networks/psp_encoder_arch.py:45
Method__init__(self, in_channels, out_channels, bias=True, bias_init_val=0, lr_mul=1, activation=None)
networks/prior_arch.py:149
Method__init__(self, in_nc, out_nc, nf, nb, gc=32, sf=4, bias=True)
networks/rrdbnet2_arch.py:53
Method__init__(self, WEncoderPath=None, PriorModelPath=None, SRModelPath=None, YoloPath=None, device='cuda')
models/TextEnhancement.py:55