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

解决utils引用冲突

parent 645b616c
No related branches found
No related tags found
No related merge requests found
File moved
import os
import mindspore
from mindspore import context
from mindspore.context import ParallelMode
import mindspore.nn as nn
import mindspore.dataset as ds
from PIL import Image
from mindspore.train.callback import ModelCheckpoint, CheckpointConfig
from mindspore.nn import Accuracy
from mindspore.train.callback import TimeMonitor, LossMonitor
from mindspore import Model
from unet_medical.unet_model import UNetMedical
from nets.deeplab_v3 import deeplab_v3
from dataset import GetDatasetGenerator
from loss import SoftmaxCrossEntropyLoss
import utils
from learning_rates import exponential_lr
context.set_context(mode=context.PYNATIVE_MODE, save_graphs=False,
device_target='Ascend', device_id=7)
......@@ -25,7 +20,7 @@ train_dataset_generator = GetDatasetGenerator('./datasets', 'train')
train_dataset = ds.GeneratorDataset(train_dataset_generator, ["data", "label"], shuffle=True)
train_dataset = train_dataset.batch(4, drop_remainder=True)
lr_iter = utils.learning_rates.exponential_lr(3e-5, 20, 0.98, 500, staircase=True)
lr_iter = exponential_lr(3e-5, 20, 0.98, 500, staircase=True)
net_loss = SoftmaxCrossEntropyLoss(6, 255)
net_opt = nn.Adam(net.trainable_params(), learning_rate=lr_iter)
......
from learning_rates import cosine_lr, poly_lr, exponential_lr
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