This is a valuable job, I want to train on my own dataset. Would you like to share the core part such as Encoder, Decoder, Masker, and Demasker?