Initialize the model with some weight

zudi-lin / pytorch_connectomics

PyTorch Connectomics: segmentation toolbox for EM connectomics

http://connectomics.readthedocs.io/

MIT License

169 stars 77 forks source link

Initialize the model with some weight #143

Closed Wangchu-Xiang closed 4 months ago

Wangchu-Xiang commented 4 months ago

Hi I am curious if it's possible to initialize the model with some wight generated in previous training. I would like to try to fine tune this model for some TEM mito-segmentation. So I would like to run the training again with some ground truth on my hand but with some initial weight.

donglaiw commented 4 months ago

models: https://huggingface.co/pytc/mito/tree/main
command: python -u scripts/main.py --inference --config-base configs/MitoEM/MitoEM-Base.yaml --config-file MitoEM-BC.yaml --checkpoint mito_u3d-bc_mitoem_300k.pth.tar