From 8613abe3e0e1f7b18ebeb5b6d38ab06a55cea749 Mon Sep 17 00:00:00 2001
From: tuanio <nvatuan3@gmail.com>
Date: Fri, 13 May 2022 10:15:52 +0700
Subject: [PATCH] add nn

---
 model.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/model.py b/model.py
index 0a37c08..1c24dc2 100644
--- 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