edvr训练

想要训练edvr,在mmediting项目中进行训练
但是reds数据集太大下载不下来,我们可以选择使用稍小一点的数据集进行训练,看看流程是否走得通

1、修改文件路径

修改配置文件中的数据集路径:
在这里插入图片描述
修改该文件内lq和gt和ann文件的路径:
在这里插入图片描述

2、生成anno文件

首先训练数据除了 低分辨率、ground truth数据、还需要一个anno数据。这个数据通过该目录下的py文件来获取
在这里插入图片描述
修改文件中range的数量,根据数据集的序列个数来修改即可。
在这里插入图片描述
之后添加–root-path参数开始运行即可,便可以得到meta_info_REDS_GT.txt文件。
在这里插入图片描述

3、修改数据集的具体参数

进入对应文件夹
在这里插入图片描述
修改对应的参数即可
在这里插入图片描述

4、报错

之后在开始训练(多卡)时会遇到这样的错误
在这里插入图片描述
告诉你模型的部分计算没有存放在计算图中,所以需要重新设置参数:
进入对应文件:
在这里插入图片描述
在改文件下,将参数find_unused_parameters设为True:
在这里插入图片描述

5、训练

一切就绪,开始训练!
在这里插入图片描述

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值