Remove deprecated map_location in export.py (#13217)

Signed-off-by: Sangbum Daniel Choi <34004152+SangbumChoi@users.noreply.github.com>
This commit is contained in:
Sangbum Daniel Choi authored and GitHub committed 2024-07-25 14:02:17 +02:00
1 parent d6112173f5
commit 6deb2d75cb
1 file changed
+1 -1
+1 -1
View File
@@ -256,7 +256,7 @@ def export_torchscript(model, im, file, optimize, prefix=colorstr("TorchScript:"
# Load model
weights = 'yolov5s.pt'
device = select_device('')
model = attempt_load(weights, map_location=device)
model = attempt_load(weights, device=device)
# Example input tensor
im = torch.zeros(1, 3, 640, 640).to(device)