Upgrade LFNA

2021-05-13 13:42:33 +00:00
parent cfabd05de8
commit c2fa181bc5
3 changed files with 5 additions and 2 deletions
--- a/exps/LFNA/lfna_meta_model.py
+++ b/exps/LFNA/lfna_meta_model.py
@@ -55,6 +55,7 @@ class LFNA_Meta(super_core.SuperModule):
                    order=super_core.LayerOrder.PostNorm,
                )
            )
+        layers.append(super_core.SuperLinear(time_embedding, time_embedding))
        self.meta_corrector = super_core.SuperSequential(*layers)

        model_kwargs = dict(