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/fpfe.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

(limited to 'models/fpfe.py')

diff --git a/models/fpfe.py b/models/fpfe.py
index 29f7a5f..28a0440 100644
--- a/models/fpfe.py
+++ b/models/fpfe.py
@@ -1,5 +1,4 @@
-from torch import nn as nn
-from torch.nn import functional as F
+import torch.nn as nn
 
 from models.layers import FocalConv2d
 
-- 
cgit v1.2.3