Quellcode durchsuchen

Fix typo (#3729)

modifyDataloader
fcakyon GitHub vor 3 Jahren
Ursprung
Commit
0e2d0d54d7
Es konnte kein GPG-Schlüssel zu dieser Signatur gefunden werden GPG-Schlüssel-ID: 4AEE18F83AFDEB23
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  1. +1
    -1
      export.py

+ 1
- 1
export.py Datei anzeigen

@@ -43,7 +43,7 @@ def run(weights='./yolov5s.pt', # weights path

# Load PyTorch model
device = select_device(device)
assert not (device.type == 'cpu' and opt.half), '--half only compatible with GPU export, i.e. use --device 0'
assert not (device.type == 'cpu' and half), '--half only compatible with GPU export, i.e. use --device 0'
model = attempt_load(weights, map_location=device) # load FP32 model
labels = model.names


Laden…
Abbrechen
Speichern