jd730 / OICR-pytorch

Pytorch Implementation of Multiple Instance Detection Network with Online Instance Classifier Refinement
MIT License
75 stars 10 forks source link

My code stopped here for a long time, is this right? #19

Closed ashencode closed 4 years ago

ashencode commented 4 years ago

My code stopped here for a long time,about 3 days, is this right?How much time does it take to train?

/home/yesenmao/shenjt/OICR-pytorch/lib/model/oicr/oicr.py(37)WeightedSoftmaxWithLoss() -> loss = (labels_ic torch.log(prob + eps)) (Pdb) > /home/yesenmao/shenjt/OICR-pytorch/lib/model/oicr/oicr.py(37)WeightedSoftmaxWithLoss() -> loss = (labels_ic torch.log(prob + eps)) (Pdb)

jd730 commented 4 years ago

Hi @ashencode , I am sorry that one pdb.set_trace() is added on https://github.com/jd730/OICR-pytorch/blob/97c8ecccfb186f35acd89f4537e5f45a245ed817/lib/model/oicr/oicr.py#L36 at 6ec17f66e65d971af0b9c5f1655aa6cfbbc99d35. Please delete that line and it will run correctly.