Skip to content

Commit 3c759d9

Browse files
committed
fix entropy lss
1 parent 21ddf5e commit 3c759d9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

trinity/algorithm/entropy_loss_fn/entropy_loss_fn.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -65,8 +65,8 @@ class DummyEntropyLossFn(EntropyLossFn):
6565
Dummy entropy loss function.
6666
"""
6767

68-
def __init__(self):
69-
pass
68+
def __init__(self, entropy_coef: float):
69+
self.entropy_coef = entropy_coef
7070

7171
def __call__(
7272
self,

0 commit comments

Comments
 (0)