Hugging Face
Models
Datasets
Spaces
Posts
Docs
Solutions
Pricing
Log In
Sign Up
Spaces:
xiexh20
/
HDM-interaction-recon
like
7
Sleeping
App
Files
Files
Community
main
HDM-interaction-recon
/
model
/
pvcnn
/
modules
/
loss.py
xiexh20
add hdm demo v1
2fd6166
4 months ago
raw
history
blame
contribute
delete
No virus
162 Bytes
import
torch.nn
as
nn
from
.
import
functional
as
F
__all__ = [
'KLLoss'
]
class
KLLoss
(nn.Module):
def
forward
(
self, x, y
):
return
F.kl_loss(x, y)