TimeGAN-pytorch icon indicating copy to clipboard operation
TimeGAN-pytorch copied to clipboard

Bugs in Discriminator (batch dimension and Linear layer output dim)

Open itakatz opened this issue 1 year ago • 1 comments

It seems to me that the implementation of the Discriminator class has 2 issues:

  1. The RNN should have the input parameter batch_first set to True, but it is not set (and defaults to False): https://github.com/zzw-zwzhang/TimeGAN-pytorch/blob/4bde0ded33d84af1504d3bc85686d10e328b8580/lib/model.py#L167
  2. The output dimension of the Linear layer should be 1 as this is a binary classification block (predicting fake or not), but it is set to the hidden dimension: https://github.com/zzw-zwzhang/TimeGAN-pytorch/blob/4bde0ded33d84af1504d3bc85686d10e328b8580/lib/model.py#L169 this does not produce an error, since later the loss function averages its input. But it means the calculation is not as it shold be.

itakatz avatar Oct 30 '23 17:10 itakatz

How long is the training time? Why do you start iterating again after 50,000 iterations, and have been cycling this iterative process of 50,000 times?

010606zss avatar Sep 18 '24 09:09 010606zss