sorry I’m still confused with that part, so far my code of that part is as follows:
net = fcn(num_classes)
net.cuda()
criterion = nn.NLLLoss2d()
basic_optim = torch.optim.SGD(net.parameters(), lr=1e-2, weight_decay=1e-4)
optimizer = ScheduledOptim(basic_optim)

may I know what’s the meaning of return the log_softmax and use NLLLoss
Thank you so much.