init

janelia-cellmap · Mar 16, 2024 · 636c416 · 636c416
1 parent 947b669
commit 636c416
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/cellmap_models/pytorch/cosem/__init__.py b/src/cellmap_models/pytorch/cosem/__init__.py
@@ -1,4 +1,5 @@
 from .load_model import load_model
+from .load_checkpoint import download_checkpoint
 
 models_dict = {
     "setup04/1820500": "https://janelia-cosem-networks.s3.amazonaws.com/v0003.2-pytorch/cosem_models/cosem_models/setup04/1820500",