diff --git a/research/cv/RCAN/script/run_ascend_standalone.sh b/research/cv/RCAN/script/run_ascend_standalone.sh index e7b45fe0d5f0f511d25d7d9e01d0ff9fde772b1a..0946ed7593c7dec8210a51205182ad176311b445 100644 --- a/research/cv/RCAN/script/run_ascend_standalone.sh +++ b/research/cv/RCAN/script/run_ascend_standalone.sh @@ -48,7 +48,7 @@ env >env.log nohup python train.py \ --batch_size 16 \ --lr 1e-4 \ - --scale 2+3+4 \ + --scale 2 \ --task_id 0 \ --dir_data $PATH1 \ --epochs 500 \ diff --git a/research/cv/RCAN/src/rcan_model.py b/research/cv/RCAN/src/rcan_model.py index 49517596ca36c1b3e0250b00fef9d5ffad56b09e..0fd5020768cfc18ef43945bcb4557f2fb31270f7 100644 --- a/research/cv/RCAN/src/rcan_model.py +++ b/research/cv/RCAN/src/rcan_model.py @@ -185,7 +185,7 @@ class RCAN(nn.Cell): n_feats = args.n_feats kernel_size = 3 reduction = args.reduction - scale = args.scale[0] + scale = args.scale self.dytpe = mstype.float16 # RGB mean for DIV2K