Project-MONAI / Project-MONAI/MONAI
Standardize the way of getting model weights from Trainer
@Warvito is already working on this.
Since Oct 21, 2022.
- Dominant language
- Python
- Stars
- 8.7k
- Forks
- 1.6k
- Avg merge
- 5d 1h
- Merged PRs (30d)
- 20
Description
Is your feature request related to a problem? Please describe.
Certain trainers, like GanTrainer use several models, while the standard SupervisedTrainer only uses one model.
Describe the solution you'd like
There should be a standardized API (e.g., get_weights/model_states) that returns all model parameters. This would be useful to allow integration with other components, like MonaiAlgo for federated learning.
Having a standardized API would allow MonaiAlgo to easily support bundles that use Trainers other than SupervisedTrainer, e.g. mednist_gan.
Describe alternatives you've considered
n/a
Additional context
n/a
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.