gradcheck will fail if input is modified in place.

vincentqb · vincentqb · commit 80534acf37df · 2021-05-27T11:11:14.000-07:00
diff --git a/test/torchaudio_unittest/rnnt/autograd_impl.py b/test/torchaudio_unittest/rnnt/autograd_impl.py
@@ -56,7 +56,7 @@ def test_RNNTLoss_gradcheck(self, data_func):
             data["logit_lengths"],
             data["target_lengths"],
         )
-        loss = RNNTLoss(blank=data["blank"])
+        loss = RNNTLoss(blank=data["blank"], reuse_logits_for_grads=False)
 
         self.assert_grad(loss, inputs, enable_all_grad=False)
 

Original file line number	Diff line number	Diff line change
`@@ -56,7 +56,7 @@ def test_RNNTLoss_gradcheck(self, data_func):`
`56`	`56`	`data["logit_lengths"],`
`57`	`57`	`data["target_lengths"],`
`58`	`58`	`)`
`59`		`- loss = RNNTLoss(blank=data["blank"])`
	`59`	`+ loss = RNNTLoss(blank=data["blank"], reuse_logits_for_grads=False)`
`60`	`60`
`61`	`61`	`self.assert_grad(loss, inputs, enable_all_grad=False)`
`62`	`62`