biobb_pytorch/mdae/models/gmvae.py
2
W504:
line break after binary operator
101
-
self
.
log_normal
(
x
,
xm
,
xv
)
+
# Reconstruction Loss
102
self
.
log_normal
(
z
,
zm
,
zv
)
-
self
.
log_normal
(
z
,
zm_prior
,
zv_prior
)
-
# Regularization Loss (KL Divergence)