Skip to content
Snippets Groups Projects
Unverified Commit 06d88c07 authored by i-robot's avatar i-robot Committed by Gitee
Browse files

!3816 modify network RCAN scripts

Merge pull request !3816 from 邓卓/modify-networks
parents 170ca7e7 31a96b56
No related branches found
No related tags found
No related merge requests found
......@@ -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 \
......
......@@ -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
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment