Skip to content

Commit

Permalink
Cellpose: fix documentation for torch parameter
Browse files Browse the repository at this point in the history
According to the __init__ signature (line 90), the default value for torch=True
  • Loading branch information
imagejan committed Mar 11, 2021
1 parent 3ac46b2 commit 305979e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion cellpose/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ class Cellpose():
where model is saved (e.g. mx.gpu() or mx.cpu()), overrides gpu input,
recommended if you want to use a specific GPU (e.g. mx.gpu(4) or torch.cuda.device(4))
torch: bool (optional, default False)
torch: bool (optional, default True)
run model using torch if available
"""
Expand Down

0 comments on commit 305979e

Please sign in to comment.