Skip to content

Commit

Permalink
fix ModelManager import
Browse files Browse the repository at this point in the history
  • Loading branch information
klwetstone committed Apr 26, 2024
1 parent 70e79f2 commit a4e75fb
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 a4e75fb

Please sign in to comment.