Method__init__(self, in_size, gate_size, inter_size, nonlocal_mode, sub_sample_factor)
code/networks/attention_unet.py:114
Method__init__(self, config, img_size=224, num_classes=21843, zero_head=False, vis=False)
code/networks/vision_transformer.py:25
Functioncross_entropy2d(logit, target, ignore_index=255, weight=None, size_average=True, batch_average=True)
code/dataloaders/utils.py:123
Functiondefine_D(input_nc, ndf, which_model_netD,
n_layers_D=3, norm='batch', use_sigmoid=False, init_type='norma
code/networks/networks_other.py:167
Functiondefine_G(input_nc, output_nc, ngf, which_model_netG, norm='batch', use_dropout=False, init_type='normal', gpu_ids=[])
code/networks/networks_other.py:143