From ed8388448149e2093011cb7cba5b82d30115fc1d Mon Sep 17 00:00:00 2001
From: seekingdeep <muharraq@outlook.com>
Date: Sun, 24 Jan 2021 01:04:47 +0300
Subject: [PATCH] Update u2net_train.py

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

diff --git a/u2net_train.py b/u2net_train.py
index 40395e9..5fe6090 100644
--- a/u2net_train.py
+++ b/u2net_train.py
@@ -1,3 +1,4 @@
+import os
 import torch
 import torchvision
 from torch.autograd import Variable