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

调整学习率改变策略

parent 373f1bef
No related merge requests found
......@@ -9,7 +9,7 @@ from mindspore import Model
from nets.deeplab_v3 import deeplab_v3
from dataset import GetDatasetGenerator
from loss import SoftmaxCrossEntropyLoss
from learning_rates import exponential_lr
from learning_rates import exponential_lr, cosine_lr
context.set_context(mode=context.PYNATIVE_MODE, save_graphs=False,
device_target='Ascend', device_id=7)
......@@ -20,7 +20,8 @@ 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 = exponential_lr(3e-5, 100, 0.98, 500, staircase=True)
# lr_iter = exponential_lr(3e-5, 100, 0.98, 500, staircase=True)
lr_iter = cosine_lr(3e-5, 20, 500)
net_loss = SoftmaxCrossEntropyLoss(6, 255)
net_opt = nn.Adam(net.trainable_params(), learning_rate=lr_iter)
......
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