Update P5 + P6 model ensembling (#3082)
This commit is contained in:
1 parent
d2a17289c9
commit
251aeafcb1
1 file changed
+3
-2
@@ -130,7 +130,8 @@ def attempt_load(weights, map_location=None, inplace=True):
|
|||||||
if len(model) == 1:
|
if len(model) == 1:
|
||||||
return model[-1] # return model
|
return model[-1] # return model
|
||||||
else:
|
else:
|
||||||
print('Ensemble created with %s\n' % weights)
|
print(f'Ensemble created with {weights}\n')
|
||||||
for k in ['names', 'stride']:
|
for k in ['names']:
|
||||||
setattr(model, k, getattr(model[-1], k))
|
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
|
return model # return ensemble
|
||||||
Reference in new issue
Block a user