iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix models

Browse Source
This commit is contained in:
D-X-Y
2021-05-26 03:25:55 +00:00
parent 50f4f3dda7
commit 33a8e7c88d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
exps/GeMOSA/lfna_meta_model.py
View File

@@ -241,7 +241,7 @@ class MetaModelV1(super_core.SuperModule):
_, [_], time_embed = self(timestamp.view(1, 1), None) _, [_], time_embed = self(timestamp.view(1, 1), None)
match_loss = criterion(new_param, time_embed) 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) y_hat = base_model.forward_with_container(x, container)
meta_loss = criterion(y_hat, y) meta_loss = criterion(y_hat, y)
loss = meta_loss + match_loss loss = meta_loss + match_loss