Method__init__(self, d_model, nhead, dim_feedforward=2048, dropout=0.1,
activation="relu", normalize_before
models/transformer.py:126
Method__init__(self, d_model, nhead, dim_feedforward=2048, dropout=0.1,
activation="relu", normalize_before
models/transformer.py:186
Functionget_pen_loss(z_pi, z_mu1, z_mu2, z_sigma1, z_sigma2, z_corr, z_pen_logits, x1_data, x2_data,
pen_data)
models/loss.py:96