from local

This commit is contained in:
dohe0342 2023-01-09 20:21:47 +09:00
parent 8b59cb1ac0
commit cc5c3eff53
3 changed files with 6 additions and 5 deletions

View File

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