From 69eb38e9bc3ecfffae834492aced5121b876feea Mon Sep 17 00:00:00 2001 From: VectorSL Date: Tue, 3 Nov 2020 18:31:27 +0800 Subject: [PATCH] update modelzoo alexnet --- model_zoo/official/cv/alexnet/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_zoo/official/cv/alexnet/train.py b/model_zoo/official/cv/alexnet/train.py index 67b45e1344..016d02d6ac 100644 --- a/model_zoo/official/cv/alexnet/train.py +++ b/model_zoo/official/cv/alexnet/train.py @@ -76,8 +76,8 @@ if __name__ == "__main__": gradients_mean=True) init() elif device_target == "GPU": - init() if device_num > 1: + init() context.reset_auto_parallel_context() context.set_auto_parallel_context(device_num=device_num, parallel_mode=ParallelMode.DATA_PARALLEL, gradients_mean=True)