From c7acb01a49af4c2529a51a35b0d2b054cd6ad74c Mon Sep 17 00:00:00 2001 From: deepr <hexiangdong2020@outlook.com> Date: Thu, 15 Jul 2021 22:44:51 +0800 Subject: [PATCH] =?UTF-8?q?Fix=20Bug:=20=E4=BF=AE=E6=94=B9=E8=AE=BE?= =?UTF-8?q?=E5=A4=87ID=E4=B8=BA7?= 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 a4989ab..f13b0a5 100644 --- a/main.py +++ b/main.py @@ -15,7 +15,7 @@ from dataset import GetDatasetGenerator from loss import SoftmaxCrossEntropyLoss context.set_context(mode=context.PYNATIVE_MODE, save_graphs=False, - device_target='Ascend', device_id=8) + device_target='Ascend', device_id=7) net = UNetMedical(n_channels=3, n_classes=6) train_dataset_generator = GetDatasetGenerator('./datasets', 'train') -- GitLab