You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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:
It seems to me that the implementation of the Discriminator class has 2 issues:
batch_first
set toTrue
, but it is not set (and defaults toFalse
):TimeGAN-pytorch/lib/model.py
Line 167 in 4bde0de
1
as this is a binary classification block (predicting fake or not), but it is set to the hidden dimension:TimeGAN-pytorch/lib/model.py
Line 169 in 4bde0de
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.
The text was updated successfully, but these errors were encountered: