Skip to content

Commit ded3124

Browse files
committed
removed detector logging in forwards
1 parent 4f7b5e1 commit ded3124

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

partinet/DynamicDet/models/yolo.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -244,7 +244,7 @@ def forward_once(self, x, profile=False):
244244
need_first_head = self.training or (self.dynamic and score[:, 0] >= self.dy_thres)
245245

246246
if need_second:
247-
print(f"This micrograph has used both detectors\n")
247+
# print(f"This micrograph has used both detectors\n")
248248
for m in self.model_b2:
249249
if m.f == 'input':
250250
x = input_x
@@ -267,7 +267,7 @@ def forward_once(self, x, profile=False):
267267
y.append(x if m.i in self.save_b2 else None) # save output
268268

269269
if need_first_head:
270-
print(f"This micrograph has used only one detector\n")
270+
# print(f"This micrograph has used only one detector\n")
271271
for m in self.model_h:
272272
if m.f != -1: # if not from previous layer
273273
x = y[m.f] if isinstance(m.f, int) else [x if j == -1 else y[j] for j in m.f] # from earlier layers

0 commit comments

Comments
 (0)