Freeze parameters for inference with `.freeze()` - Githubissues

cellarium-ai / cellarium-ml

Distributed single-cell data analysis.

BSD 3-Clause "New" or "Revised" License

11 stars 2 forks source link

Freeze parameters for inference with `.freeze()` #186

Closed ordabayevy closed 2 months ago

ordabayevy commented 4 months ago

https://lightning.ai/docs/pytorch/stable/common/lightning_module.html#freeze