You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+9-20Lines changed: 9 additions & 20 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -84,30 +84,19 @@ Pass `hubs=` to track model download provenance. Hubs are framework-agnostic and
84
84
|`huggingface`| Downloads via `huggingface_hub`|
85
85
|`torchhub`| Downloads via `torch.hub`|
86
86
87
-
For `torch` and `keras`, models are user-defined subclasses so there's no constructor to patch. Use `client.load()` to get load/unload tracking alongside inference:
88
-
89
-
```python
90
-
model = client.load(MyModel)
91
-
output = model(x) # tracked automatically
92
-
```
93
-
94
87
For unsupported frameworks, see [Manual tracking](docs/manual-tracking.md).
Copy file name to clipboardExpand all lines: docs/manual-tracking.md
+9Lines changed: 9 additions & 0 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -9,6 +9,15 @@ Use manual tracking when your framework is not covered by a WildEdge integration
9
9
- You want to attach input/output metadata (token counts, image dimensions, confidence scores, etc.)
10
10
- You want to record user feedback tied to a specific inference
11
11
12
+
## torch and keras
13
+
14
+
For `torch` and `keras`, models are user-defined subclasses with no constructor to patch. Use `client.load()` to get load, unload, and inference tracking automatically:
15
+
16
+
```python
17
+
model = client.load(MyModel)
18
+
output = model(x) # tracked automatically
19
+
```
20
+
12
21
## Register your model
13
22
14
23
Every model needs a handle before you can track events against it. Pass the model object and an explicit `model_id`:
0 commit comments