Hugging Face
Models
Datasets
Spaces
Posts
Docs
Solutions
Pricing
Log In
Sign Up
Spaces:
xiexh20
/
HDM-interaction-recon
like
8
Sleeping
App
Files
Files
Community
0732f1c
HDM-interaction-recon
/
model
/
pvcnn
/
modules
/
loss.py
xiexh20
add hdm demo v1
2fd6166
6 months ago
raw
Copy download link
history
blame
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)