Skip to content
Snippets Groups Projects
Commit 31a96b56 authored by yuxiaodianyu's avatar yuxiaodianyu
Browse files

modify network RCAN scripts

parent 03f7c052
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