kwotsin / mimicry

[CVPR 2020 Workshop] A PyTorch GAN library that reproduces research results for popular GANs.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Did update code base without tensorflow ??

Johnson-yue opened this issue · comments

Hi, I see your code, including many tensorflow used ,such as
compute_fidcompute_kid, compute_inception_score, and so on

Would you plan to update pytorch code for those function??

Hi @Johnson-yue, indeed TensorFlow is used quite a bit for the metrics implementation, mainly to make sure the scores can be reproduced as closely to original as possible. For example as mentioned in #7, inception score can be quite sensitive to weight differences when using a different model, and the tensorflow inception model most people use for research seems to be using a very particular implementation (e.g. the output node has >1k outputs).

However, indeed the tensorflow use seems to be quite bulky compared to a pure pytorch approach. Recently the move to TF2 also has some breaking changes that I'm hoping to update and wait to see if it these changes remain stable in the long run. Hopefully there won't be a need to update too much whenever TF updates!