!8945 remove network useless config parameters.

From: @linqingke
Reviewed-by: @yingjy,@oacjiewen
Signed-off-by: @yingjy
pull/8945/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 364c01d677

@ -169,7 +169,6 @@ bash scripts/run_eval_ascend.sh $TRAINED_CKPT
* `--TRAIN_DATASET_SIZE`:训练数据集大小。 * `--TRAIN_DATASET_SIZE`:训练数据集大小。
* `--TEST_DATASET_PATH`:测试数据集的路径。 * `--TEST_DATASET_PATH`:测试数据集的路径。
* `--TEST_BATCH_SIZE`:测试批次大小。 * `--TEST_BATCH_SIZE`:测试批次大小。
* `--TEST_DATASET_SIZE`:测试数据集大小。
* `--TRAIN_EPOCHS`:总训练轮次。 * `--TRAIN_EPOCHS`:总训练轮次。
* `--CKPT_PATH`:模型检查点文件路径,可用于恢复训练和评估。 * `--CKPT_PATH`:模型检查点文件路径,可用于恢复训练和评估。
* `--SAVE_PATH`:模型检查点文件保存路径。 * `--SAVE_PATH`:模型检查点文件保存路径。

@ -22,7 +22,6 @@ config = ed({
"img_height": 768, "img_height": 768,
"keep_ratio": False, "keep_ratio": False,
"flip_ratio": 0.5, "flip_ratio": 0.5,
"photo_ratio": 0.5,
"expand_ratio": 1.0, "expand_ratio": 1.0,
# anchor # anchor
@ -109,7 +108,6 @@ config = ed({
"test_max_per_img": 100, "test_max_per_img": 100,
"test_batch_size": 2, "test_batch_size": 2,
"rpn_head_loss_type": "CrossEntropyLoss",
"rpn_head_use_sigmoid": True, "rpn_head_use_sigmoid": True,
"rpn_head_weight": 1.0, "rpn_head_weight": 1.0,
@ -118,7 +116,6 @@ config = ed({
"base_step": 58633, "base_step": 58633,
"total_epoch": 13, "total_epoch": 13,
"warmup_step": 500, "warmup_step": 500,
"warmup_mode": "linear",
"warmup_ratio": 1/3.0, "warmup_ratio": 1/3.0,
"sgd_step": [8, 11], "sgd_step": [8, 11],
"sgd_momentum": 0.9, "sgd_momentum": 0.9,

@ -39,7 +39,6 @@ config = ed({
"TRAIN_ROOT_DIR": 'psenet/ic15/', "TRAIN_ROOT_DIR": 'psenet/ic15/',
"TRAIN_IS_TRANSFORM": True, "TRAIN_IS_TRANSFORM": True,
"TRAIN_LONG_SIZE": 640, "TRAIN_LONG_SIZE": 640,
"TRAIN_DATASET_SIZE": 1000,
"TRAIN_MIN_SCALE": 0.4, "TRAIN_MIN_SCALE": 0.4,
"TRAIN_BATCH_SIZE": 4, "TRAIN_BATCH_SIZE": 4,
"TRAIN_REPEAT_NUM": 1800, "TRAIN_REPEAT_NUM": 1800,
@ -48,7 +47,6 @@ config = ed({
# dataset for test # dataset for test
"TEST_ROOT_DIR": 'psenet/ic15/', "TEST_ROOT_DIR": 'psenet/ic15/',
"TEST_DATASET_SIZE": 500,
"TEST_BUFFER_SIZE": 4, "TEST_BUFFER_SIZE": 4,
"TEST_DROP_REMAINDER": False, "TEST_DROP_REMAINDER": False,

@ -14,7 +14,6 @@
# ============================================================================ # ============================================================================
cfg_unet = { cfg_unet = {
'name': 'Unet',
'lr': 0.0001, 'lr': 0.0001,
'epochs': 400, 'epochs': 400,
'distribute_epochs': 1600, 'distribute_epochs': 1600,

Loading…
Cancel
Save