修改训练参数

main.py

@@ -28,10 +28,10 @@ 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, 20, 0.9, 100, staircase=True)
+# lr_iter = exponential_lr(3e-5, 20, 0.9, 100, staircase=True)
 
 net_loss = SoftmaxCrossEntropyLoss(6, 255)
-net_opt = nn.Adam(net.trainable_params(), learning_rate=lr_iter)
+net_opt = nn.Adam(net.trainable_params(), learning_rate=3e-4)
 
 config_ck = CheckpointConfig(save_checkpoint_steps=30, keep_checkpoint_max=1)
 ckpoint = ModelCheckpoint(prefix="deeplab.ckpt", config=config_ck)