add nn

tuanio · May 13, 2022 · 8613abe · 8613abe
1 parent a6ae761
commit 8613abe
Showing 1 changed file with 1 addition and 0 deletions.
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