Skip to content

Commit 651bd0c

Browse files
authored
[rl] stop adding checkpoint paths to metrics (#69)
1 parent 529b73f commit 651bd0c

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tinker_cookbook/supervised/train.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -211,14 +211,13 @@ async def finish_batch(submitted: SubmittedBatch):
211211

212212
if submitted.step % config.save_every == 0 and submitted.step > 0:
213213
with timed("save_checkpoint", metrics):
214-
checkpoint_paths = await checkpoint_utils.save_checkpoint_async(
214+
await checkpoint_utils.save_checkpoint_async(
215215
training_client=training_client,
216216
name=f"{submitted.step:06d}",
217217
log_path=config.log_path,
218218
loop_state={"epoch": submitted.epoch_idx, "batch": submitted.batch_idx},
219219
kind="both",
220220
)
221-
metrics.update(checkpoint_paths)
222221

223222
with timed("step", metrics):
224223
fwd_bwd_result = await submitted.fwd_bwd_future.result_async()

0 commit comments

Comments
 (0)