From 90dd0828bf46eb10579fa1e5318d55b51be54458 Mon Sep 17 00:00:00 2001
From: gengdongjie <gengdongjie@huawei.com>
Date: Sat, 9 Oct 2021 15:03:06 +0800
Subject: [PATCH] fix bugs for cyclegan and advanced_east
---
research/cv/CycleGAN/export.py | 3 +--
research/cv/advanced_east/README.md | 2 +-
.../cv/advanced_east/scripts/run_standalone_train_ascend.sh | 4 ++--
3 files changed, 4 insertions(+), 5 deletions(-)
diff --git a/research/cv/CycleGAN/export.py b/research/cv/CycleGAN/export.py
index a6e10b528..c17337047 100644
--- a/research/cv/CycleGAN/export.py
+++ b/research/cv/CycleGAN/export.py
@@ -16,14 +16,13 @@
"""export file."""
import numpy as np
-from mindspore import context, Tensor
+from mindspore import Tensor
from mindspore.train.serialization import export
from src.models.cycle_gan import get_generator
from src.utils.args import get_args
from src.utils.tools import load_ckpt
args = get_args("export")
-context.set_context(mode=context.GRAPH_MODE, device_target="Ascend", device_id=args.device_id)
if __name__ == '__main__':
G_A = get_generator(args)
diff --git a/research/cv/advanced_east/README.md b/research/cv/advanced_east/README.md
index ac4e477c2..380d5105a 100644
--- a/research/cv/advanced_east/README.md
+++ b/research/cv/advanced_east/README.md
@@ -245,7 +245,7 @@ The Normalization of model training on Ascend is GN, the model training on GPU i
| Resources | Ascend 910 | Tesla V100S-PCIE 32G|
| MindSpore Version | 1.1 |1.1 |
| Dataset | MTWI-2018 |MTWI-2018 |
-| Training Parameters | epoch=18, batch_size = 8, lr=1e-3 |epoch=84, batch_size = 8, lr=1e-3 |
+| Training Parameters | epoch=6, batch_size=2, lr=1e-4 |epoch=6, batch_size=2, lr=1e-3 |
| Optimizer | AdamWeightDecay |AdamWeightDecay |
| Loss Function | QuadLoss |QuadLoss |
| Outputs | matrix with size of 3x64x64,3x96x96,3x112x112 |matrix with size of 3x64x64,3x96x96,3x112x112 |
diff --git a/research/cv/advanced_east/scripts/run_standalone_train_ascend.sh b/research/cv/advanced_east/scripts/run_standalone_train_ascend.sh
index 0b42fb53d..2baa72b83 100644
--- a/research/cv/advanced_east/scripts/run_standalone_train_ascend.sh
+++ b/research/cv/advanced_east/scripts/run_standalone_train_ascend.sh
@@ -16,9 +16,9 @@
echo "=============================================================================================================="
echo "Please run the script as: "
-echo "bash run_standalone_train_ascend.sh"
+echo "bash run_standalone_train_ascend.sh [dataset_path](optional)"
echo "for example: bash run_standalone_train_ascend.sh"
echo "=============================================================================================================="
python train_single_size.py \
- --device_target="Ascend" > output.train.log 2>&1 &
+ --device_target="Ascend" --data_path $1 > output.train.log 2>&1 &
--
GitLab