Rollback ctcLibDir variable assignment when CTC_LIB is defined

上级 8494d579
...@@ -23,7 +23,9 @@ class ConnectionistTemporalClassification(gof.COp): ...@@ -23,7 +23,9 @@ class ConnectionistTemporalClassification(gof.COp):
if self.computeGradient: if self.computeGradient:
self.gradients = T.ftensor3(name="ctc_grad") self.gradients = T.ftensor3(name="ctc_grad")
if not "CTC_LIB" in os.environ: if "CTC_LIB" in os.environ:
self.ctcLibDir = os.environ["CTC_LIB"]
else:
raise EnvironmentError("CTC_LIB environment variable is not set.") raise EnvironmentError("CTC_LIB environment variable is not set.")
def c_lib_dirs(self): def c_lib_dirs(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论