Fixing loading path of noise flow model.

BorealisAI · Sep 11, 2019 · 0e8d9a4 · 0e8d9a4
1 parent 017610a
commit 0e8d9a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/borealisflows/NoiseFlowWrapper.py b/borealisflows/NoiseFlowWrapper.py
@@ -39,7 +39,7 @@ def __init__(self, path):
         self.temp = 1.0
 
         self.hps = self.hps_loader(os.path.join(self.nf_path, 'hps.txt'))
-        self.ckpt_dir = os.path.join(self.hps.logdir, 'ckpt')
+        self.ckpt_dir = os.path.join(self.nf_path, 'ckpt')
         self.model_checkpoint_path = os.path.join(self.ckpt_dir, 'model.ckpt.best')
         self.load_noise_flow_model()