search

bfortuner / pytorch_tiramisu

FC-DenseNet in PyTorch for Semantic Segmentation
MIT License
306 stars 66 forks source link

Need help for test. #11

Closed CocoRLin closed 6 years ago

CocoRLin commented 6 years ago

Hello! I'm new for neural network and pytorch! When I run this code on my own dataset , I have confusion about test() in train.py

  1. I use model.load_state_dict( torch.load('weights-100-0.018-0.065.pth')) to load parameters before train_utils.test(model, test_loader, criterion, epoch=1)
    but i have error like this: KeyError: 'unexpected key "startEpoch" in state_dict'
  2. How to use latest.th?

Thank you for your help!

GWwangshuo commented 6 years ago

@CocoRLin I encountered the same problem. Have you solved ? Thanks