LegoFormer icon indicating copy to clipboard operation
LegoFormer copied to clipboard

Can you publish the training code?

Open wangleishan opened this issue 3 years ago • 3 comments

I've been reading your article recently, but the training code is missing. Can you publish the training code?Thank you!

wangleishan avatar Aug 25 '21 09:08 wangleishan

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.

sharlingw avatar Aug 31 '21 03:08 sharlingw

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?

wangleishan avatar Sep 01 '21 09:09 wangleishan

Facing the same requirement as @wangleishan. Trying to reproduce but not so much success so far.

azfaruzman avatar Sep 11 '21 16:09 azfaruzman