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 53834b7 commit 7595c45Copy full SHA for 7595c45
applications/ColossalChat/coati/trainer/dpo.py
@@ -382,7 +382,7 @@ def _criterion(outputs, inputs):
382
)
383
self.accumulative_meter.reset()
384
self.num_train_step += 1
385
-
+
386
if self.save_dir is not None and self.num_train_step > 0 and self.num_train_step % self.save_interval == 0:
387
# save checkpoint
388
self.coordinator.print_on_master("\nStart saving model checkpoint with running states")
0 commit comments