From 71575036e54ad1c2b1bf4c543718337ba84ee0de Mon Sep 17 00:00:00 2001
From: Xiaoyang Guo <xyguo@ee.cuhk.edu.hk>
Date: Sun, 14 Apr 2019 09:29:58 -0400
Subject: [PATCH] fix import bug

---
 main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main.py b/main.py
index beb9f4d..5b6cd80 100644
--- a/main.py
+++ b/main.py
@@ -14,7 +14,7 @@ import numpy as np
 import time
 from tensorboardX import SummaryWriter
 from datasets import __datasets__
-from models import __models__
+from models import __models__, model_loss
 from utils import *
 from torch.utils.data import DataLoader
 import gc