add benckmark

2026-09-24 23:33:08 +08:00 · 2022-07-30 10:00:11 +08:00
parent f6250d61e4
commit dd1e17fd79
28 changed files with 425 additions and 59 deletions
@@ -157,6 +157,7 @@ def main(config, device, logger, vdl_writer):
        scaler = paddle.amp.GradScaler(
            init_loss_scaling=scale_loss,
            use_dynamic_loss_scaling=use_dynamic_loss_scaling)
+        model, optimizer = paddle.amp.decorate(models=model, optimizers=optimizer, level='O2', master_weight=True)
    else:
        scaler = None