Add miscellaneous updates (#8)

2023-03-13 13:48:38 -07:00
parent e9d3f2ff77
commit cfae35b861
7 changed files with 44 additions and 22 deletions
--- a/cacheflow/models/sample.py
+++ b/cacheflow/models/sample.py
@@ -11,7 +11,7 @@ from cacheflow.sequence import SequenceOutputs
 class Sampler(nn.Module):

    def __init__(self) -> None:
-        super().__init__()
+        super(Sampler, self).__init__()

    def forward(
        self,