Skip to content

Commit

Permalink
Update main.py (#2742)
Browse files Browse the repository at this point in the history
### Changes

Slight redundancy in the code 

### Reason for changes

<!--- Why should the change be applied -->

May reduce confusion 

<!--- Post the numerical ID of the ticket, if available -->

### Tests

<!--- How was the correctness of changes tested and whether new tests
were added -->
  • Loading branch information
Lars-Codes authored Jun 19, 2024
1 parent 25399bd commit 58b0c64
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions examples/post_training_quantization/openvino/yolov8/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,8 +73,6 @@ def prepare_validation(model: YOLO, args: Any) -> Tuple[Validator, torch.utils.d

data_loader = validator.get_dataloader(f"{DATASETS_DIR}/coco128", 1)

validator = model.smart_load("validator")(args)

validator.is_coco = True
validator.class_map = coco80_to_coco91_class()
validator.names = model.model.names
Expand Down

0 comments on commit 58b0c64

Please sign in to comment.