Update drn_train.py

fix some bug about train DRN model
own
kongdebug 3 years ago committed by GitHub
parent d29af2909c
commit b2d22d0c89
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 5
      tutorials/train/image_restoration/drn_train.py

@ -5,8 +5,6 @@ sys.path.append(os.path.abspath('../PaddleRS'))
import paddle import paddle
import paddlers as pdrs import paddlers as pdrs
if __name__ == "__main__":
# 定义训练和验证时的transforms # 定义训练和验证时的transforms
train_transforms = pdrs.datasets.ComposeTrans( train_transforms = pdrs.datasets.ComposeTrans(
input_keys=['lq', 'gt'], input_keys=['lq', 'gt'],
@ -78,4 +76,5 @@ if __name__ == "__main__":
output_dir='output_dir', output_dir='output_dir',
validate=5000, validate=5000,
snapshot=5000, snapshot=5000,
lr_rate=0.0001) lr_rate=0.0001,
log=10)

Loading…
Cancel
Save