diff --git a/main.py b/main.py index a4989abf58e12a13cad8568c9a6859427dfd0a56..f13b0a55e863480465f63555142a2dfd502d78bf 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')