PU-GAN icon indicating copy to clipboard operation
PU-GAN copied to clipboard

Question about the pretrain model

Open satyremusique opened this issue 3 years ago • 0 comments

Thank you for this great work. Just a simple question, is the pretrain model that you share was train with exactly the code in the github. And is is the same(same hyperparameter) than the article.

I am able to use that pretrain model to do inference, but I am not able to use it to fine-tune with different data. It's seem that the miss match of tensor in the discriminator(when given the output of the generator). But if I retrain from scratch and save result, I seem to be able to fine tune it after that.

satyremusique avatar Mar 19 '21 13:03 satyremusique