From 33a8e7c88da1bff525d780021cb4e66af50b75b4 Mon Sep 17 00:00:00 2001
From: D-X-Y <280835372@qq.com>
Date: Wed, 26 May 2021 03:25:55 +0000
Subject: [PATCH] Fix models

---
 exps/GeMOSA/lfna_meta_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/exps/GeMOSA/lfna_meta_model.py b/exps/GeMOSA/lfna_meta_model.py
index 2ef4286..9c69c83 100644
--- 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