Update LFNA version 1.0

2021-05-07 14:27:15 +08:00
parent 80aaac4dfa
commit 34560ad8d1
5 changed files with 120 additions and 40 deletions
--- a/exps/LFNA/basic-his.py
+++ b/exps/LFNA/basic-his.py
@@ -86,9 +86,10 @@ def main(args):
            input_dim=1,
            output_dim=1,
            act_cls="leaky_relu",
-            norm_cls="simple_norm",
-            mean=mean,
-            std=std,
+            norm_cls="identity",
+            # norm_cls="simple_norm",
+            # mean=mean,
+            # std=std,
        )
        model = get_model(dict(model_type="simple_mlp"), **model_kwargs)
        # build optimizer