LegoFormer
LegoFormer copied to clipboard
Can you publish the training code?
I've been reading your article recently, but the training code is missing. Can you publish the training code?Thank you!
I think a change to eval.py will do. Instead of loading the model from the checkpoints folder, initialize it yourself and then just add a line of training code using pytorch-lightening.
I think a change to eval.py will do. Instead of loading the model from the checkpoints folder, initialize it yourself and then just add a line of training code using pytorch-lightening.
My coding ability is relatively poor, but I want to reproduce this paper, can you help me give the training code?
Facing the same requirement as @wangleishan. Trying to reproduce but not so much success so far.