How should we train the model using tf 2.6.0?
xiaogangzhu opened this issue · 4 comments
Hi, I tried training the model on tf 2.6.0 and some problems happens. How should i fix this?
Currently we are not planning to create a branch that supports TF 2.6.0.
However, you can check one of our recent works which utilizes TF 2.6.0, where we used similar Coarse and Fine generators, and Discriminator with a embedding feature-matching loss. The overall training is quite similar, and should be reproducible for RVGAN.
Hope this helps !
Is your recent work VTGAN works for vessel segmentation? I want to try the model on my own retinal pictures and it seems there is only pretrained model so I need to train by myself. But unfortunately the scripts don't work for my RTX3090 lol. I was wondering if I can get the final model weight file and it will be really helpful!
I have created a new branch for tf 2.6.0
Link: https://github.com/SharifAmit/RVGAN/tree/tf-2.6
Please open a new issue if you have any error.
Thanks!
Thank you!