Skip to content

Commit ee5fd22

Browse files
committed
Implement method using torch.compile with customized backend.
1 parent f159a3d commit ee5fd22

File tree

2 files changed

+217
-128
lines changed

2 files changed

+217
-128
lines changed

graph_net/paddle/validate.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,8 +36,6 @@ def _extract_forward_source(model_path, class_name):
3636
source = f.read()
3737

3838
tree = ast.parse(source)
39-
forward_code = None
40-
4139
for node in tree.body:
4240
if isinstance(node, ast.ClassDef) and node.name == class_name:
4341
for fn in node.body:

0 commit comments

Comments
 (0)