Skip to content

Commit 7595c45

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 53834b7 commit 7595c45

File tree

1 file changed

+1
-1
lines changed
  • applications/ColossalChat/coati/trainer

1 file changed

+1
-1
lines changed

applications/ColossalChat/coati/trainer/dpo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -382,7 +382,7 @@ def _criterion(outputs, inputs):
382382
)
383383
self.accumulative_meter.reset()
384384
self.num_train_step += 1
385-
385+
386386
if self.save_dir is not None and self.num_train_step > 0 and self.num_train_step % self.save_interval == 0:
387387
# save checkpoint
388388
self.coordinator.print_on_master("\nStart saving model checkpoint with running states")

0 commit comments

Comments
 (0)