From b609951561979bf1e6344336d4e4776e43c4217d Mon Sep 17 00:00:00 2001
From: Marcin <marcin.kardas@gmail.com>
Date: Thu, 14 Feb 2019 00:00:56 +0100
Subject: [PATCH] Fix path of pretrained model

---
 ulmfit/pretrain_lm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ulmfit/pretrain_lm.py b/ulmfit/pretrain_lm.py
index 61a81ad..b89a994 100644
--- a/ulmfit/pretrain_lm.py
+++ b/ulmfit/pretrain_lm.py
@@ -214,7 +214,7 @@ class LMHyperParams:
             learn.freeze()
         if self.pretrained_fnames is not None:
             print("Loading pretrained model")
-            fnames = [learn.path/learn.model_dir/f'{fn}.{ext}' for fn,ext in zip(self.pretrained_fnames, ['pth', 'pkl'])]
+            fnames = [f'{fn}.{ext}' for fn,ext in zip(self.pretrained_fnames, ['pth', 'pkl'])]
             learn.load_pretrained(*fnames)
             learn.freeze()
         # compared to standard Adam, we set beta_1 to 0.8