Skip to content

Commit

Permalink
Fix typo in multi-species docs
Browse files Browse the repository at this point in the history
  • Loading branch information
ethanwhite committed Dec 12, 2023
1 parent 7593b54 commit 6fc3475
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/multi_species.md
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ To load the backboard and box prediction portions of the release model, but crea
Here is an example using the alive/dead tree data stored in the package, but the same logic applies to the bird detector.

```
m = main.deepforest(num_classes=2, label_dict={"Alive":0,"Dead":0})
m = main.deepforest(num_classes=2, label_dict={"Alive":0,"Dead":1})
deepforest_release_model = main.deepforest()
deepforest_release_model.use_release()
m.model.backbone.load_state_dict(deepforest_release_model.model.backbone.state_dict())
Expand Down

0 comments on commit 6fc3475

Please sign in to comment.