We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d83dc33 commit bf56a44Copy full SHA for bf56a44
graph_net/torch/graph_variable_renamer.py
@@ -88,7 +88,7 @@ def __call__(self, rel_model_path):
88
return
89
90
src_model_path = os.path.join(self.config["model_path_prefix"], rel_model_path)
91
- with cuda_gc(enabled=self.config["release_gpu_memory"]):
+ with cuda_gc():
92
module, inputs = get_torch_module_and_inputs(src_model_path)
93
gm = parse_sole_graph_module(module, inputs)
94
gm, rename_map = self.rename_graph_variables(gm, inputs, src_model_path)
0 commit comments