fixed the segmentation relabel bug and pre-trained model loader bug

zudi-lin / pytorch_connectomics

PyTorch Connectomics: segmentation toolbox for EM connectomics

http://connectomics.readthedocs.io/

MIT License

171 stars 77 forks source link

fixed the segmentation relabel bug and pre-trained model loader bug #38

Closed divyam-goel closed 4 years ago

zudi-lin commented 4 years ago

hi @divyam-goel, could you please reverse the commit related to --checkpoint (commit c8c9d19)? I am designing a new logic to handle checkpoint I/O and the original model loading functions in connectomics/model/__init__.py will be deprecated. I will merge the fix of the relabeling bugs after your revision.

divyam-goel commented 4 years ago

hello @zudi-lin, I have reverted the checkpoint change. Please merge the relabel bug fix.