Skip to content

P
pytensor
提交 58f083bd authored 作者: Frederic's avatar Frederic
浏览文件
操作

Fix access to variable.

上级 3849399c
隐藏空白字符变更
内嵌 并排
正在显示 包含 2 行增加 和 2 行删除
theano/gof/vm.py
浏览文件 @ 58f083bd
...@@ -319,7 +319,7 @@ class Stack(VM): ...@@ -319,7 +319,7 @@ class Stack(VM):
s=o[0].size s=o[0].size
dtype = str(o[0].dtype) dtype = str(o[0].dtype)
dtype2 = dtype[-3:] dtype2 = dtype[-3:]
s *= memory_size_map[dtype2] # KeyError here: couldn't determine the dtype memory size s *= self.memory_size_map[dtype2] # KeyError here: couldn't determine the dtype memory size
size.append(s) size.append(s)
self.outputs_size[current_apply] = size self.outputs_size[current_apply] = size
except Exception: except Exception:
...@@ -370,7 +370,7 @@ class Stack(VM): ...@@ -370,7 +370,7 @@ class Stack(VM):
s=o[0].size s=o[0].size
dtype = str(o[0].dtype) dtype = str(o[0].dtype)
dtype2 = dtype[-2:] dtype2 = dtype[-2:]
s *= memory_size_map[dtype2] # KeyError here: couldn't determine the dtype memory size s *= self.memory_size_map[dtype2] # KeyError here: couldn't determine the dtype memory size
size.append(s) size.append(s)
self.outputs_size[current_apply] = size self.outputs_size[current_apply] = size
if self.allow_gc: if self.allow_gc:
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论