zhejiang-lab
/
ts-model-refining
Not watched
1
0
Fork 0
Code
Releases 0 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: 361cc9256f
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '361cc9256f'
${ noResults }
ts-model-refining/OneFlow-KAE/kamal/core/tasks/loss/loss.py

8 lines
193 B
Raw Blame History 

  1. from .functional import *

  2. 

  3. class KLDiv(object):

  4.     def __init__(self, T=1.0):

  5.         self.T = T

  6.     

  7.     def __call__(self, logits, targets):

  8.         return kldiv( logits, targets, T=self.T )