STEGO
STEGO copied to clipboard
no module named 'core'
Hello,Hi, thanks for sharing your code! I met with a question when I ran eval_segmention.py.How to deal with it?Thank you!
Me too! Have you fix it?
In the previous code. The src/utils.py was called src/core.py. I think that is the origin of the problem.
I avoid this error by changing the "run_picie" from True to False in the src/configs/eval_config.yml. (The picie is another unsupervised segmentation method compared in this work)
@MY-LIU100101 thanks for finding the fix for this issue, I will double check on my side to see whether this is still happening on master
The inappropriate model-saving approach might cause this issue. https://github.com/pytorch/pytorch/issues/18325#issuecomment-661759359 Saving only the state dict instead of the whole model object is more recommended. https://pytorch.org/docs/master/notes/serialization.html#recommended-approach-for-saving-a-model