From fbbc6a2dc533c480059dd8b621ff5abef349b593 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Thu, 24 Dec 2020 18:51:33 +0800
Subject: Optimize imports

---
 models/tfa.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'models/tfa.py')

diff --git a/models/tfa.py b/models/tfa.py
index 72bfcfb..b80328a 100644
--- a/models/tfa.py
+++ b/models/tfa.py
@@ -1,7 +1,7 @@
 import copy
 
 import torch
-from torch import nn as nn
+import torch.nn as nn
 
 from models.layers import BasicConv1d
 
-- 
cgit v1.2.3