Updates

2021-05-27 14:09:52 +00:00
parent 726dbef326
commit 30f78f87d2
1 changed files with 1 additions and 1 deletions
--- a/exps/GeMOSA/meta_model_ablation.py
+++ b/exps/GeMOSA/meta_model_ablation.py
@@ -237,7 +237,7 @@ class MetaModel_TraditionalAtt(super_core.SuperModule):
            for iepoch in range(epochs):
                optimizer.zero_grad()
                time_embed = self.gen_time_embed(timestamp.view(1))
-                match_loss = criterion(new_param, time_embed)
+                match_loss = F.l1_loss(new_param, time_embed)

                [container] = self.gen_model(new_param.view(1, -1))
                y_hat = base_model.forward_with_container(x, container)