zc911/MatrixSlow

matrixslow/core/core.py中的一处疑问

XuanchenLi opened this issue · 1 comments

image
此处的
return [node for node in default_graph.nodes if isinstance(node, Variable) and node.trainable]
是否应该改为:
return [node for node in graph.nodes if isinstance(node, Variable) and node.trainable]

您提得对。能否发一个 Merge Request ,我们合并进来。