From 56e666690a0865b0a4a00c62757069fe2ff2ad9a Mon Sep 17 00:00:00 2001 From: deepr <hexiangdong2020@outlook.com> Date: Tue, 20 Jul 2021 23:03:50 +0800 Subject: [PATCH] =?UTF-8?q?=E8=A7=A3=E5=86=B3utils=E5=BC=95=E7=94=A8?= =?UTF-8?q?=E5=86=B2=E7=AA=81?= 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 2a920a4..a349975 100644 --- a/main.py +++ b/main.py @@ -14,7 +14,7 @@ from unet_medical.unet_model import UNetMedical from nets.deeplab_v3 import deeplab_v3 from dataset import GetDatasetGenerator from loss import SoftmaxCrossEntropyLoss -import utils.learning_rates as learning_rates +from utils import learning_rates context.set_context(mode=context.PYNATIVE_MODE, save_graphs=False, device_target='Ascend', device_id=7) -- GitLab