hanq

Perhaps you can try to change "result_dict = net(inputs, gt, prefix="val")" in train.py file to "result_dict = net.module.forward(inputs, gt, prefix="val")"

me too