Git Product home page Git Product logo

Comments (9)

DLwbm123 avatar DLwbm123 commented on July 28, 2024

Same to me.

from edvr.

xinntao avatar xinntao commented on July 28, 2024

Yes, indeed we also found the training with DCN is unstable.
We will write down the issues we met during the competition in this repo later. And unstable training is one of them.
There are still a lot of things that we can improve on EDVR and we are also exploring some of them.

During the competition, we trained the large model from smaller ones and used a smaller learning rate for dcn. Even with these tricks, the over-large offsets are occasionally met. And we just resumed it from a normal checkpoint if we met.

from edvr.

huihuiustc avatar huihuiustc commented on July 28, 2024

What dou mean that you trained the large model from smaller ones.Or this one:"We initialize deeper networks by parameters from shallower ones for faster convergence"in your paper.

For instance. We use kaiming_normal initialize all parameter,then freeze TSA and Reconstruction Module,only request_grad in the PCD align and PreDeblur Module.

Thanks for your attention.

from edvr.

xinntao avatar xinntao commented on July 28, 2024
  1. Yes, we first train shallower ones.
  2. We will release some models and also the training codes to train from scratch. But their performances are not as good as the competition models.

from edvr.

huihuiustc avatar huihuiustc commented on July 28, 2024

谢谢大佬的回复,确实是很厉害的工作和研究。

我们正在尝试先把可变卷积换成正常的卷积,然后训练得到的初始model,然后用这个模型训练网络。
接着冻结部分模型块再开始训练。

from edvr.

xinntao avatar xinntao commented on July 28, 2024

Actually, DCN is relatively important. So you can first train a small network with DCN (w/o TSA).
We are running these experiments and will release it as soon as possible.

from edvr.

splinter21 avatar splinter21 commented on July 28, 2024

1、“We trained the large model from smaller ones and used a smaller learning rate for dcn.”
Do you mean this(for example):
step 1>5front-10back with DCN+TSA,lr=1e-4,(model S(hallow)).
step 2>5front-40back with DCN+TSA,lr(DCN)=5e-5(e.g.),lr_others=1e-4. And parameters of S except 30 back blocks is copied to model D(eep).

2、"You can first train a small network with DCN (w/o TSA)"
Do you mean, only DCN is needed to be pretrained, another paramters after DCN is not needed(not useful for deeper model).
For example, I can train 5front blocks with DCN, w/o TSA, and with very shallow SR network after DCN.
Then, the DCN is pretrained, paramters after DCN can be abandoned, and I can change SR network whatever I like after DCN?

3、This pretrained-DCN-trick can't make the final model D with a deeper or wider(I mean, change the feature extraction layers before DCN) DCN module compared with model S, because DCN paramters are needed to be copied. Is it right?

4、For the second step, there are two choices for DCN. The first one, smaller lr for DCN. The second one, freeze DCN module. The second choice can save many time and GPU memory for training. Is it suitable?
@xinntao

from edvr.

xinntao avatar xinntao commented on July 28, 2024

We have updated the training codes and configs. We provide training scripts for the model with Channel=128, Back RB=10.
The learning rate scheme is different from that in the competition. But it is more effective.

  1. train with the script train_EDVR_woTSA_M.yml
  2. then train with the script train_EDVR_M.yml

You can try this.

from edvr.

tongjuntx avatar tongjuntx commented on July 28, 2024

谢谢大佬的回复,确实是很厉害的工作和研究。

我们正在尝试先把可变卷积换成正常的卷积,然后训练得到的初始model,然后用这个模型训练网络。
接着冻结部分模型块再开始训练。

have you succeed?how about the effect?

from edvr.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.