From 68cb7e4b27741f7f380404a42e5973292613ce33 Mon Sep 17 00:00:00 2001
From: Marco De Nadai <me@marcodena.it>
Date: Wed, 16 Sep 2020 19:49:16 +0200
Subject: [PATCH] Bugfixes

---
 lpips/trainer.py | 4 +++-
 requirements.txt | 1 +
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/lpips/trainer.py b/lpips/trainer.py
index 4c77c7be..1d171356 100755
--- a/lpips/trainer.py
+++ b/lpips/trainer.py
@@ -10,6 +10,8 @@
 from tqdm import tqdm
 from IPython import embed
 import lpips
+import os
+
 
 class Trainer():
     def name(self):
@@ -63,7 +65,7 @@ def initialize(self, model='lpips', net='alex', colorspace='Lab', pnet_rand=Fals
 
         if self.is_train: # training mode
             # extra network on top to go from distances (d0,d1) => predicted human judgment (h*)
-            self.rankLoss = networks.BCERankingLoss()
+            self.rankLoss = lpips.BCERankingLoss()
             self.parameters += list(self.rankLoss.net.parameters())
             self.lr = lr
             self.old_lr = lr
diff --git a/requirements.txt b/requirements.txt
index 9d1fa5dc..c17b42c2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,3 +6,4 @@ scikit-image>=0.13.0
 opencv>=2.4.11
 matplotlib>=1.5.1
 tqdm>=4.28.1
+jupyter