add nn

tuanio · tuanio · commit 8613abe3e0e1 · 2022-05-13T10:15:52.000+07:00
diff --git a/model.py b/model.py
@@ -1,4 +1,5 @@
 import torch
+from torch import nn
 from torchaudio.models import DeepSpeech
 import pytorch_lightning as pl
 from utils import TextProcess, CTCDecoder

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`import torch`
	`2`	`+from torch import nn`
`2`	`3`	`from torchaudio.models import DeepSpeech`
`3`	`4`	`import pytorch_lightning as pl`
`4`	`5`	`from utils import TextProcess, CTCDecoder`