iicd/autodl-projects
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'main' of github.com:D-X-Y/AutoDL-Projects

Browse Source
This commit is contained in:
D-X-Y
2021-05-27 23:04:38 +08:00
parent 9a5663390a 30f78f87d2
commit 0070e54869
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
exps/GeMOSA/meta_model_ablation.py
View File

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