Skip to content

Commit

Permalink
fix formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
ilyes319 committed Jan 16, 2025
1 parent c7adcbf commit e0f7a5b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mace/cli/run_train.py
Original file line number Diff line number Diff line change
Expand Up @@ -646,7 +646,7 @@ def run(args: argparse.Namespace) -> None:
distributed_model = DDP(model, device_ids=[local_rank])
else:
distributed_model = None

tools.train(
model=model,
loss_fn=loss_fn,
Expand Down

0 comments on commit e0f7a5b

Please sign in to comment.