Skip to content

Commit bf56a44

Browse files
committed
Minor fix
1 parent d83dc33 commit bf56a44

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

graph_net/torch/graph_variable_renamer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ def __call__(self, rel_model_path):
8888
return
8989

9090
src_model_path = os.path.join(self.config["model_path_prefix"], rel_model_path)
91-
with cuda_gc(enabled=self.config["release_gpu_memory"]):
91+
with cuda_gc():
9292
module, inputs = get_torch_module_and_inputs(src_model_path)
9393
gm = parse_sole_graph_module(module, inputs)
9494
gm, rename_map = self.rename_graph_variables(gm, inputs, src_model_path)

0 commit comments

Comments
 (0)