Training By Myself

[Hank5120]

Hello,
I want to train model by myself,
But I don't know which dataset you use and the train code you use.
Will you release this part?
thanks

[Hank5120]

I add another image imput, but it will error because the channel size is different,
how can I do to adjust this part?