Skip to content
Snippets Groups Projects
Commit c7acb01a authored by 何向东's avatar 何向东
Browse files

Fix Bug: 修改设备ID为7

parent 03215f75
No related branches found
No related tags found
No related merge requests found
...@@ -15,7 +15,7 @@ from dataset import GetDatasetGenerator ...@@ -15,7 +15,7 @@ from dataset import GetDatasetGenerator
from loss import SoftmaxCrossEntropyLoss from loss import SoftmaxCrossEntropyLoss
context.set_context(mode=context.PYNATIVE_MODE, save_graphs=False, 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) net = UNetMedical(n_channels=3, n_classes=6)
train_dataset_generator = GetDatasetGenerator('./datasets', 'train') train_dataset_generator = GetDatasetGenerator('./datasets', 'train')
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment