Skip to content

Commit fe5b4a7

Browse files
call cancel() only in IEModel class (#3449)
1 parent d3ddd45 commit fe5b4a7

File tree

2 files changed

+9
-4
lines changed

2 files changed

+9
-4
lines changed

demos/action_recognition_demo/python/action_recognition_demo/models.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -117,6 +117,10 @@ def wait_request(self, req_id):
117117
self.infer_queue[req_id].wait()
118118
return self.outputs.pop(req_id, None)
119119

120+
def cancel(self):
121+
for ireq in self.infer_queue:
122+
ireq.cancel()
123+
120124

121125
class DummyDecoder:
122126
def __init__(self, num_requests=2):
@@ -133,3 +137,6 @@ def async_infer(self, model_input, req_id):
133137
def wait_request(self, req_id):
134138
assert req_id in self.requests
135139
return self.requests.pop(req_id)
140+
141+
def cancel(self):
142+
pass

demos/action_recognition_demo/python/action_recognition_demo/steps.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -102,8 +102,7 @@ def __init__(self, encoder):
102102
self.async_model = AsyncWrapper(self.encoder, self.encoder.num_requests)
103103

104104
def __del__(self):
105-
for ireq in self.encoder.infer_queue:
106-
ireq.cancel()
105+
self.encoder.cancel()
107106

108107
def process(self, frame):
109108
preprocessed = preprocess_frame(frame)
@@ -126,8 +125,7 @@ def __init__(self, decoder, sequence_size=16):
126125
self._embeddings = deque(maxlen=self.sequence_size)
127126

128127
def __del__(self):
129-
for ireq in self.decoder.infer_queue:
130-
ireq.cancel()
128+
self.decoder.cancel()
131129

132130
def process(self, item):
133131
if item is None:

0 commit comments

Comments
 (0)