fix import bug

2019-04-14 09:29:58 -04:00 · 2019-04-14 09:29:58 -04:00 · 71575036e5
commit 71575036e5
parent 9f079c6505
1 changed files with 1 additions and 1 deletions
--- 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