Merge pull request #386 from bigict/optim

chungongyu · web-flow · commit f1206a20ae13 · 2026-01-28T12:49:11.000+08:00
refactor: code reformat
diff --git a/profold2/command/trainer.py b/profold2/command/trainer.py
@@ -343,7 +343,8 @@ def _step(data_loader, it, writer, stage='train', batch_callback=None):
     optimizer.zero_grad(set_to_none=True)
 
     logging.debug(
-        '_step it: %d, loss_scaler: %f, lr: %s', it, loss_scaler, scheduler.get_lr()
+        '_step it: %d, loss_scaler: %f, lr: %s', it, loss_scaler,
+        scheduler.get_last_lr()
     )
 
     running_loss = MetricDict()

Original file line number	Diff line number	Diff line change
`@@ -343,7 +343,8 @@ def _step(data_loader, it, writer, stage='train', batch_callback=None):`
`343`	`343`	`optimizer.zero_grad(set_to_none=True)`
`344`	`344`
`345`	`345`	`logging.debug(`
`346`		`- '_step it: %d, loss_scaler: %f, lr: %s', it, loss_scaler, scheduler.get_lr()`
	`346`	`+ '_step it: %d, loss_scaler: %f, lr: %s', it, loss_scaler,`
	`347`	`+ scheduler.get_last_lr()`
`347`	`348`	`)`
`348`	`349`
`349`	`350`	`running_loss = MetricDict()`