Skip to content

Commit

Permalink
fix ModelManager import
Browse files Browse the repository at this point in the history
  • Loading branch information
klwetstone authored and ejm714 committed May 17, 2024
1 parent da686dd commit 8f5233d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/docs/yaml-config.md
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@ Only some of the possible parameters can be passed directly as arguments to the
The main API for zamba is the [`ModelManager` class](api-reference/models-model_manager.md#zamba.models.model_manager.ModelManager) that can be accessed with:

```python
from zamba.models.manager import ModelManager
from zamba.models.model_manager import ModelManager
```

The `ModelManager` class is used by `zamba`’s command line interface to handle preprocessing the filenames, loading the videos, training the model, performing inference, and saving predictions. Therefore any functionality available to the command line interface is accessible via the `ModelManager` class.
Expand Down

0 comments on commit 8f5233d

Please sign in to comment.