This commit is contained in:
D-X-Y 2021-05-27 14:09:52 +00:00
parent 726dbef326
commit 30f78f87d2

View File

@ -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)