thinking-machines-lab/tinker
1
0
Fork 0
mirror of https://github.com/thinking-machines-lab/tinker.git synced 2026-04-19 12:58:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(docs): fwdbwd_result.loss does not exist, use metrics[loss:sum]

Browse source 
Made-with: Cursor
This commit is contained in:
Shrinav Loka 2026-04-17 00:01:24 -07:00
parent 2a1f5e3240
commit 3446d62f27
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
docs/api/trainingclient.md
View file

@ -108,7 +108,7 @@ optim_future = training_client.optim_step(
)
fwdbwd_result = await fwdbwd_future
print(f"Loss: {fwdbwd_result.loss}")
print(f"Loss: {fwdbwd_result.metrics['loss:sum']}")
```
#### `forward_backward_async`

2
src/tinker/lib/public_interfaces/training_client.py
View file

@ -294,7 +294,7 @@ class TrainingClient(TelemetryProvider):
)
fwdbwd_result = await fwdbwd_future
print(f"Loss: {fwdbwd_result.loss}")
print(f"Loss: {fwdbwd_result.metrics['loss:sum']}")
```
"""
requests = self._chunked_requests(data)