from local

This commit is contained in:
dohe0342 2023-01-09 19:53:11 +09:00
parent f41acc98ef
commit 5eb0639b1c
2 changed files with 1 additions and 1 deletions

View File

@ -272,4 +272,4 @@ class Interformer(nn.Module):
)
'''
return (simple_loss, pruned_loss)
return loss