Merge branch 'main' of github.com:D-X-Y/AutoDL-Projects
This commit is contained in:
commit
0070e54869
@ -237,7 +237,7 @@ class MetaModel_TraditionalAtt(super_core.SuperModule):
|
|||||||
for iepoch in range(epochs):
|
for iepoch in range(epochs):
|
||||||
optimizer.zero_grad()
|
optimizer.zero_grad()
|
||||||
time_embed = self.gen_time_embed(timestamp.view(1))
|
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))
|
[container] = self.gen_model(new_param.view(1, -1))
|
||||||
y_hat = base_model.forward_with_container(x, container)
|
y_hat = base_model.forward_with_container(x, container)
|
||||||
|
Loading…
Reference in New Issue
Block a user