From 251aeafcb16ebc4c9d9a6641b3677aaac2f2d2cb Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 8 May 2021 19:06:12 +0200 Subject: [PATCH] Update P5 + P6 model ensembling (#3082) --- models/experimental.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/models/experimental.py b/models/experimental.py index 85a22d42fa54..afa787907104 100644 --- a/models/experimental.py +++ b/models/experimental.py @@ -130,7 +130,8 @@ def attempt_load(weights, map_location=None, inplace=True): if len(model) == 1: return model[-1] # return model else: - print('Ensemble created with %s\n' % weights) - for k in ['names', 'stride']: + print(f'Ensemble created with {weights}\n') + for k in ['names']: setattr(model, k, getattr(model[-1], k)) + model.stride = model[torch.argmax(torch.tensor([m.stride.max() for m in model])).int()].stride # max stride return model # return ensemble