mirror of
https://github.com/thinking-machines-lab/tinker.git
synced 2026-04-22 16:49:00 +00:00
Sync contents
This commit is contained in:
parent
5ad4282c96
commit
e7a0d0ca2d
15 changed files with 91 additions and 31 deletions
|
|
@ -21,7 +21,7 @@ Args:
|
|||
|
||||
Example:
|
||||
```python
|
||||
training_client = service_client.create_lora_training_client(base_model="Qwen/Qwen2.5-7B")
|
||||
training_client = service_client.create_lora_training_client(base_model="Qwen/Qwen3-8B")
|
||||
fwdbwd_future = training_client.forward_backward(training_data, "cross_entropy")
|
||||
optim_future = training_client.optim_step(types.AdamParams(learning_rate=1e-4))
|
||||
fwdbwd_result = fwdbwd_future.result() # Wait for gradients
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue