VAR icon indicating copy to clipboard operation
VAR copied to clipboard
verified publisher icon FoundationVision

training code

Open Jiawei-Yang opened this issue 1 year ago ā€¢ 3 comments

Hi there,

thanks for your amazing work! I noticed the recent updates covered the training code. Will there also be updates about training scripts and data preparation? I'm particularly interested in training the VQVAE part.

Jiawei-Yang avatar Apr 15 '24 02:04 Jiawei-Yang

Jiawei-Yang

@Jiawei-Yang thanks for your kindly attention, we will release the training code of VQVAE in several days

enjoyyi00 avatar Apr 15 '24 03:04 enjoyyi00

Jiawei-Yang

@Jiawei-Yang thanks for your kindly attention, we will release the training code of VQVAE in several days

Hello, I am interested in replicating your vqvae results and would appreciate some additional details to help me understand and possibly extend your work. Could you please provide more specifics regarding the training configuration? e.g. exact loss function equation used during training and weights for each term, batch size, lr, etc. Thanks for your good work!

shliu0 avatar Apr 16 '24 15:04 shliu0

Also very interested in the training of VQ-tokenizer part, looking forward! BTW: VAR is a super cool work!

pkulwj1994 avatar Apr 18 '24 09:04 pkulwj1994