From 3a4938dd8c787a5078a49c5f9fb5b9dab31430bc Mon Sep 17 00:00:00 2001 From: linqingke Date: Tue, 24 Nov 2020 15:32:45 +0800 Subject: [PATCH] remove network useless config parameters. --- model_zoo/official/cv/cnnctc/README_CN.md | 1 - model_zoo/official/cv/faster_rcnn/src/config.py | 3 --- model_zoo/official/cv/psenet/src/config.py | 2 -- model_zoo/official/cv/unet/src/config.py | 1 - 4 files changed, 7 deletions(-) diff --git a/model_zoo/official/cv/cnnctc/README_CN.md b/model_zoo/official/cv/cnnctc/README_CN.md index 6e8fe58aee..e09ad96c9f 100644 --- a/model_zoo/official/cv/cnnctc/README_CN.md +++ b/model_zoo/official/cv/cnnctc/README_CN.md @@ -169,7 +169,6 @@ bash scripts/run_eval_ascend.sh $TRAINED_CKPT * `--TRAIN_DATASET_SIZE`:训练数据集大小。 * `--TEST_DATASET_PATH`:测试数据集的路径。 * `--TEST_BATCH_SIZE`:测试批次大小。 - * `--TEST_DATASET_SIZE`:测试数据集大小。 * `--TRAIN_EPOCHS`:总训练轮次。 * `--CKPT_PATH`:模型检查点文件路径,可用于恢复训练和评估。 * `--SAVE_PATH`:模型检查点文件保存路径。 diff --git a/model_zoo/official/cv/faster_rcnn/src/config.py b/model_zoo/official/cv/faster_rcnn/src/config.py index 28123399e8..a2a63947fb 100644 --- a/model_zoo/official/cv/faster_rcnn/src/config.py +++ b/model_zoo/official/cv/faster_rcnn/src/config.py @@ -22,7 +22,6 @@ config = ed({ "img_height": 768, "keep_ratio": False, "flip_ratio": 0.5, - "photo_ratio": 0.5, "expand_ratio": 1.0, # anchor @@ -109,7 +108,6 @@ config = ed({ "test_max_per_img": 100, "test_batch_size": 2, - "rpn_head_loss_type": "CrossEntropyLoss", "rpn_head_use_sigmoid": True, "rpn_head_weight": 1.0, @@ -118,7 +116,6 @@ config = ed({ "base_step": 58633, "total_epoch": 13, "warmup_step": 500, - "warmup_mode": "linear", "warmup_ratio": 1/3.0, "sgd_step": [8, 11], "sgd_momentum": 0.9, diff --git a/model_zoo/official/cv/psenet/src/config.py b/model_zoo/official/cv/psenet/src/config.py index 8798fa8045..e6aec4c0b3 100644 --- a/model_zoo/official/cv/psenet/src/config.py +++ b/model_zoo/official/cv/psenet/src/config.py @@ -39,7 +39,6 @@ config = ed({ "TRAIN_ROOT_DIR": 'psenet/ic15/', "TRAIN_IS_TRANSFORM": True, "TRAIN_LONG_SIZE": 640, - "TRAIN_DATASET_SIZE": 1000, "TRAIN_MIN_SCALE": 0.4, "TRAIN_BATCH_SIZE": 4, "TRAIN_REPEAT_NUM": 1800, @@ -48,7 +47,6 @@ config = ed({ # dataset for test "TEST_ROOT_DIR": 'psenet/ic15/', - "TEST_DATASET_SIZE": 500, "TEST_BUFFER_SIZE": 4, "TEST_DROP_REMAINDER": False, diff --git a/model_zoo/official/cv/unet/src/config.py b/model_zoo/official/cv/unet/src/config.py index 6946eafca3..9f8eba7b45 100644 --- a/model_zoo/official/cv/unet/src/config.py +++ b/model_zoo/official/cv/unet/src/config.py @@ -14,7 +14,6 @@ # ============================================================================ cfg_unet = { - 'name': 'Unet', 'lr': 0.0001, 'epochs': 400, 'distribute_epochs': 1600,