fix bugs for cyclegan and advanced_east

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)

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       |

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 &