Pārlūkot izejas kodu

Update P5 + P6 model ensembling (#3082)

modifyDataloader
Glenn Jocher GitHub pirms 3 gadiem
vecāks
revīzija
251aeafcb1
Šim parakstam datu bāzē netika atrasta zināma atslēga GPG atslēgas ID: 4AEE18F83AFDEB23
1 mainītis faili ar 3 papildinājumiem un 2 dzēšanām
  1. +3
    -2
      models/experimental.py

+ 3
- 2
models/experimental.py Parādīt failu

@@ -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

Notiek ielāde…
Atcelt
Saglabāt