diff --git a/research/cv/CycleGAN/export.py b/research/cv/CycleGAN/export.py index a6e10b528cd13e2edb79bc2a69a93a702f71e429..c17337047b3e97e0148d222663c5fb1b1c7a8aae 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 ac4e477c2d0caf7e9a40e46fd6e8b6e5c23c3cc3..380d5105a6569b0679d81b8080a68dd8c4c89e2a 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 0b42fb53d9c52540e7571fc157f21dbfb14df02c..2baa72b83ec782e72af8399583bcaa28ef2667d1 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 &