latent_dim not right?

[xiaolongwu0713]

Hi, thanks for sharing the code. However, the excution failed using the mitbih data with error message complaining the dimension mismatch.
I think the latent_dim when initialize gen_net should be args.latent_dim instead of 100.
gen_net = Generator(seq_len=187, channels=1, num_classes=5, latent_dim=100, data_embed_dim=10,
label_embed_dim=10 ,depth=3, num_heads=5,
forward_drop_rate=0.5, attn_drop_rate=0.5)

Could you please help to confirm this?

[Xiaomin-Li]

Hi, the args.latent_dim was set to 100 in all my experiments. So they are both correct.