↓ 7 callersMethod__init__(self, input_dim, out_dim, fc_dim, n_fc,
weight_norm=False, activation='relu', normalize_mlp=
model/transformer_vae_batch.py:14
FunctionVAECELoss(v_pre, v_gt, f, logvar, mu, beta=1e-2, lambd=1, lambd_s=1, loss='cham_smooth')
loss/loss.py:18
Method__init__(self, dim_in=3, points=10000, seq_len=50, c_dim=12, z_dim=32, ff_size=1024, num_layers=4, num_heads=4, dropou
model/transformer_vae_batch.py:107
Method__init__(self, dim_in=3, points=10000, seq_len=50,
c_dim=16, z_dim=32, ff_size=1024, num_layers=4, n
model/transformer_vae_batch.py:210
Method__init__(self, dim_in=3,
dim_h=128, z_dim = 32,
c_dim_in=12, c_dim=16,
model/transformer_vae_batch.py:285