local_plot predictions always blank about rvgan HOT 3 CLOSED

Hi,

We have not tested or experimented on tf-2.6 gpu version for our model. Please use the original tf-2.0 gpu implementation to replicate our experiment.

This version was created in response to this issue

All I can suggest is to train for longer for g_local_model to get good output. As It was mentioned in the paper to train for 100 epochs in three stages to get good output. So You need to resume training by loading weights for both g_local and g_global after 100th epoch. And in total you need to do it for 300 epochs (3 stages).

Hope this answers your questions.

from rvgan.

Thanks. Yes, I'm getting much better results with tf-2.0gpu. Seeing segmentation in both local and global.

from rvgan.

Thanks for the update. I am closing the issue for now.

Please reopen this issue if any other problem happens.

from rvgan.