fix bug in creating training weights directory

gkane26 · May 8, 2021 · f83840c · f83840c
1 parent eda25fb
commit f83840c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/psychrnn/backend/rnn.py b/psychrnn/backend/rnn.py
@@ -500,7 +500,7 @@ def train(self, trial_batch_generator, train_params={}):
         # --------------------------------------------------
         if training_weights_path != None:
             if path.dirname(training_weights_path) != "" and not path.exists(path.dirname(training_weights_path)):
-                makedirs(path.dirname(_weights_path))
+                makedirs(path.dirname(training_weights_path))
 
         # --------------------------------------------------
         # Compute gradients