Skip to content

Commit

Permalink
Update import paths for READMEs
Browse files Browse the repository at this point in the history
  • Loading branch information
rhoadesScholar committed Aug 13, 2024
1 parent 566ceed commit 31c528c
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion src/cellmap_models/pytorch/cellpose/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,7 @@ Once you have chosen a model based on the descriptions above, you can download i
If you would like to load a model for your own use, you can do the following:

```python
from cellmap_models.cellpose import load_model
from cellmap_models.pytorch.cellpose import load_model
model = load_model('<model_name>')
```

Expand Down
4 changes: 2 additions & 2 deletions src/cellmap_models/pytorch/cosem/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ The networks have been converted to PyTorch from their original Tensorflow versi
You can load a PyTorch model using the following code:

```python
import cellmap_models.cosem as cosem_models
import cellmap_models.pytorch.cosem as cosem_models
model = cosem_models.load_model('setup04/1820500')
# The model is now ready to use
```
Expand All @@ -23,7 +23,7 @@ cosem_models.models_list
Each model has a separate backbone and single layer prediction head. The `backbone` and `head` objects are both PyTorch modules and can be used as such. You can access the separate components of the model using the following code:

```python
import cellmap_models.cosem as cosem_models
import cellmap_models.pytorch.cosem as cosem_models
model = cosem_models.load_model('setup04/1820500')
backbone = model.backbone
head = model.prediction_head
Expand Down

0 comments on commit 31c528c

Please sign in to comment.