Fix bugs in LFNA

2021-05-16 21:56:09 +10:00
parent 9a2c9fc435
commit 06fe246d82
9 changed files with 34 additions and 109 deletions
--- a/exps/LFNA/lfna_meta_model.py
+++ b/exps/LFNA/lfna_meta_model.py
@@ -75,8 +75,7 @@ class LFNA_Meta(super_core.SuperModule):

        # unknown token
        self.register_parameter(
-            "_unknown_token",
-            torch.nn.Parameter(torch.Tensor(1, time_embedding)),
+            "_unknown_token", torch.nn.Parameter(torch.Tensor(1, time_embedding)),
        )

        # initialization