Skip to content

Commit

Permalink
fix ModelManager import (#325)
Browse files Browse the repository at this point in the history
  • Loading branch information
klwetstone authored May 18, 2024
1 parent da686dd commit 223c541
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 223c541

Please sign in to comment.