(self, infer_node_idx=-1)
| 125 | self.trt_use_static = False |
| 126 | |
| 127 | def get_fetch_list(self, infer_node_idx=-1): |
| 128 | fetch_names = [ |
| 129 | var.alias_name |
| 130 | for var in list(self.model_conf.values())[infer_node_idx].fetch_var |
| 131 | ] |
| 132 | return fetch_names |
| 133 | |
| 134 | def set_max_concurrency(self, concurrency): |
| 135 | self.max_concurrency = concurrency |