PyTorch 1.6.0 compatability updates

This commit is contained in:
Glenn Jocher committed 2020-07-28 15:38:56 -07:00
1 parent 7f8471eaeb
commit 43a616a955
3 files changed
+7 -6

No files matched your search

+1 -1
View File
@@ -154,7 +154,7 @@ if __name__ == '__main__':
with torch.no_grad():
if opt.update: # update all models (to fix SourceChangeWarning)
for opt.weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt', 'yolov3-spp.pt']:
for opt.weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt']:
detect()
strip_optimizer(opt.weights)
else: