from local

This commit is contained in:
dohe0342 2023-01-09 20:31:36 +09:00
parent f007665de4
commit 1a3323e574
3 changed files with 0 additions and 1 deletions

View File

@ -247,7 +247,6 @@ class Interformer(nn.Module):
with torch.no_grad():
encoder_out, x_lens, pt_layer_outputs = self.pt_encoder(x,
x_lens,
warmup=warmup,
get_layer_output=True
)