Explorar o código

fixed alphabet loading path

Min %!s(int64=4) %!d(string=hai) anos
pai
achega
bbcf31634e
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      models/basic.py

+ 1 - 1
models/basic.py

@@ -6,7 +6,7 @@ from scipy.spatial import cKDTree
 from os import path
 import tensorflow as tf
 
-ALPHABET_DIR = "./alphabets"
+ALPHABET_DIR = path.join(path.dirname(__file__), "../alphabets")
 
 
 def load_alphabet(name, polar=True):