Procedure to load this pre-trained model?
#1
by
arjunnarayanan
- opened
Hello, I'm interested in experimenting with this pre-trained model. Do you have some documentation on how to load it?
The model weights are located in the "Files" tab.
To instantiate the model, use the architecture found in the official Tencent repository:
https://github.com/Tencent/MedicalNet/blob/master/models/resnet.py
The repository's README has a step-by-step guide on running both train.py and test.py scripts. Do note you can choose to pass either an untrained or trained model instance (by loading the provided weights using the PyTorch framework methods) when calling these scripts.
For further assistance, refeer to either the PyTorch documentation or the official TencentMedicalNet repo.
Awesome, thanks for the info!
arjunnarayanan
changed discussion status to
closed