Skip to content

Commit 6fdf5ad

Browse files
committed
update code
1 parent 660ef59 commit 6fdf5ad

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

graph_net/torch/backend/unstable_to_stable_backend.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,6 @@ def _impl_unstable_to_stable_set_grad_enabled(self, gm):
134134
def replace_in_graph(graph_mod):
135135
for node in graph_mod.graph.nodes:
136136
if node.op == "call_function":
137-
# 宽松匹配所有 set_grad_enabled 相关 target
138137
if "set_grad_enabled" in str(node.target):
139138
node.target = torch.set_grad_enabled
140139
graph_mod.recompile()

0 commit comments

Comments
 (0)