We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 529b73f commit 651bd0cCopy full SHA for 651bd0c
tinker_cookbook/supervised/train.py
@@ -211,14 +211,13 @@ async def finish_batch(submitted: SubmittedBatch):
211
212
if submitted.step % config.save_every == 0 and submitted.step > 0:
213
with timed("save_checkpoint", metrics):
214
- checkpoint_paths = await checkpoint_utils.save_checkpoint_async(
+ await checkpoint_utils.save_checkpoint_async(
215
training_client=training_client,
216
name=f"{submitted.step:06d}",
217
log_path=config.log_path,
218
loop_state={"epoch": submitted.epoch_idx, "batch": submitted.batch_idx},
219
kind="both",
220
)
221
- metrics.update(checkpoint_paths)
222
223
with timed("step", metrics):
224
fwd_bwd_result = await submitted.fwd_bwd_future.result_async()
0 commit comments