I want to load the model after saving it but i got this error
NotADirectoryError: [Errno 20] Not a directory: './module/net9-1.957-0.991.pth/'
code
import parameter as para
# Define network and load parameters test.py
net = torch.nn.DataParallel(ResUNet(training=False)).cuda()
net.load_state_dict(torch.load(para.module_path))
net.eval()
#para parameter.py
module_path = './module/net9-1.957-0.991.pth/' # Test model path