From 6235489aa4c6408ddc06dc7f3ae580c1688abbd1 Mon Sep 17 00:00:00 2001 From: deepr <hexiangdong2020@outlook.com> Date: Wed, 11 Aug 2021 16:00:36 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E8=AE=AD=E7=BB=83=E5=8F=82?= =?UTF-8?q?=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 7a7f0e5..fb62879 100644 --- a/main.py +++ b/main.py @@ -21,7 +21,7 @@ from learning_rates import exponential_lr, cosine_lr context.set_context(mode=context.PYNATIVE_MODE, save_graphs=False, - device_target='Ascend', device_id=7) + device_target='Ascend', device_id=6) # net = UNetMedical(n_channels=3, n_classes=6) net = deeplab_v3.DeepLabV3(phase='train', num_classes=6, output_stride=16, freeze_bn=False) -- GitLab