From 6aa8e3cb39fcd968c59a61b7286e8fb2914f31af Mon Sep 17 00:00:00 2001
From: Jason Dsouza <61184127+jasmcaus@users.noreply.github.com>
Date: Tue, 24 Nov 2020 16:05:49 +0530
Subject: [PATCH] torchvision's models not used

---
 model/u2net.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/model/u2net.py b/model/u2net.py
index ece59e0..1ca0c03 100644
--- a/model/u2net.py
+++ b/model/u2net.py
@@ -1,6 +1,5 @@
 import torch
 import torch.nn as nn
-from torchvision import models
 import torch.nn.functional as F
 
 class REBNCONV(nn.Module):