jiyegege / animeganv2_tensorflow2 Goto Github PK

Thank you very much for your work, I did some attempts to convert from animeganv2_tf1 to tf2 before but failed.
But there is a question I want to ask. How many epochs you have trained to achieve your training results?
My results after 400 epochs are still poor, I don't know if it's a configuration issue or something else.

cuda 11.2
cudnn 8.2.1
tensorflow 2.8

According to the generator's style loss provided by https://github.com/TachibanaYoshino/AnimeGANv2/blob/617ab10ce21877df90c9cf2b705712bc81b1e2e6/tools/ops.py
Maybe function gram() should be expressed like:

feat = tf.linalg.einsum('bijc,bijd->bcd', x, x)
num_locations = tf.math.reduce_prod(x.shape[1:])
num_locations = tf.cast(num_locations, dtype=tf.float32)

feat = feat/num_locations

Otherwise, it may cause generator gradient explosion, and difficult to achieve Antagonism effect