fix resnext50 gpu amp

4 years ago · c470c9acef
parent 8df757143a
commit c470c9acef
1 changed files with 2 additions and 6 deletions
--- a/model_zoo/official/cv/resnext50/train.py
+++ b/model_zoo/official/cv/resnext50/train.py
@ -240,12 +240,8 @@ def train(cloud_args=None):
    else:
        loss_scale_manager = FixedLossScaleManager(args.loss_scale, drop_overflow_update=False)

-    if args.platform == "Ascend":
    model = Model(network, loss_fn=loss, optimizer=opt, loss_scale_manager=loss_scale_manager,
                  metrics={'acc'}, amp_level="O3")
-    else:
-        model = Model(network, loss_fn=loss, optimizer=opt, loss_scale_manager=loss_scale_manager,
-                      metrics={'acc'}, amp_level="O2")

    # checkpoint save
    progress_cb = ProgressMonitor(args)