Commit 11c9070d authored by Dan Lyth's avatar Dan Lyth
Browse files

removing train.py from this branch

parent eaf7947b
......@@ -836,7 +836,6 @@ def main():
resume_step = None
for batch in train_dataloader:
breakpoint()
with accelerator.accumulate(model):
loss, train_metric = train_step(batch, accelerator, autocast_kwargs)
accelerator.backward(loss)
......
This diff is collapsed.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment