Fix models

2021-05-26 03:25:55 +00:00
parent 50f4f3dda7
commit 33a8e7c88d
1 changed files with 1 additions and 1 deletions
--- a/exps/GeMOSA/lfna_meta_model.py
+++ b/exps/GeMOSA/lfna_meta_model.py
@@ -241,7 +241,7 @@ class MetaModelV1(super_core.SuperModule):
                _, [_], time_embed = self(timestamp.view(1, 1), None)
                match_loss = criterion(new_param, time_embed)

-                _, [container], time_embed = self(None, new_param.view(1, 1, -1))
+                _, [container], time_embed = self(None, new_param.view(1, -1))
                y_hat = base_model.forward_with_container(x, container)
                meta_loss = criterion(y_hat, y)
                loss = meta_loss + match_loss