Update PyTorch .cpu().numpy() to .cpu().detach().numpy()

wannaphong · wannaphong · commit bcd2be442d9d · 2020-04-28T21:02:34.000+07:00
diff --git a/pythainlp/transliterate/thai2rom.py b/pythainlp/transliterate/thai2rom.py
@@ -93,7 +93,7 @@ def romanize(self, text: str) -> str:
             target_tensor = (
                     torch.argmax(
                         target_tensor_logits.squeeze(1),
-                        1).cpu().numpy()
+                        1).cpu().detach().numpy()
                 )
             target = [self._ix_to_target_char[t] for t in target_tensor]
 
diff --git a/pythainlp/transliterate/thaig2p.py b/pythainlp/transliterate/thaig2p.py
@@ -94,7 +94,7 @@ def g2p(self, text: str) -> str:
             target_tensor = (
                     torch.argmax(
                         target_tensor_logits.squeeze(1),
-                        1).cpu().numpy()
+                        1).cpu().detach().numpy()
                 )
             target = [self._ix_to_target_char[t] for t in target_tensor]
 

Original file line number	Diff line number	Diff line change
`@@ -93,7 +93,7 @@ def romanize(self, text: str) -> str:`
`93`	`93`	`target_tensor = (`
`94`	`94`	`torch.argmax(`
`95`	`95`	`target_tensor_logits.squeeze(1),`
`96`		`- 1).cpu().numpy()`
	`96`	`+ 1).cpu().detach().numpy()`
`97`	`97`	`)`
`98`	`98`	`target = [self._ix_to_target_char[t] for t in target_tensor]`
`99`	`99`
Original file line number	Diff line number	Diff line change
`@@ -94,7 +94,7 @@ def g2p(self, text: str) -> str:`
`94`	`94`	`target_tensor = (`
`95`	`95`	`torch.argmax(`
`96`	`96`	`target_tensor_logits.squeeze(1),`
`97`		`- 1).cpu().numpy()`
	`97`	`+ 1).cpu().detach().numpy()`
`98`	`98`	`)`
`99`	`99`	`target = [self._ix_to_target_char[t] for t in target_tensor]`
`100`	`100`