SBU-BMI / wsinfer

🔥 🚀 Blazingly fast pipeline for patch-based classification in whole slide images
https://wsinfer.readthedocs.io
Apache License 2.0
59 stars 10 forks source link

DOCS: add info on how to download models in singularity image #64

Closed kaczmarj closed 1 year ago

kaczmarj commented 1 year ago

singularity images are read-only, so one cannot download model weights using the default configuration. the TORCH_HOME variable points to a /var/lib directory in the container, but to download model weights in singularity, one can unset that variable to use pytorch's default, which is in ~/.cache/torch/hub/checkpoints.

singularity run --env TORCH_HOME="" ...