diff --git a/config/experiments/demo_prom/example_image_detection_yolo.yaml b/config/experiments/demo_prom/example_image_detection_yolo.yaml index e440f6af..b6ae9e21 100644 --- a/config/experiments/demo_prom/example_image_detection_yolo.yaml +++ b/config/experiments/demo_prom/example_image_detection_yolo.yaml @@ -1,7 +1,7 @@ # @package _global_ accelerator: gpu batch_size: 16 -ckpt_path: /home/qazybek/restart/innofw/dui/logs/train/rotten_fruits/example_image_detection_yolo.yaml/20240826-203849/checkpoints/best.pt +ckpt_path: https://api.blackhole.ai.innopolis.university/pretrained/prom/fruits/new_baseline.pt defaults: - override /models: detection/yolov5fruits_pretrain - override /datasets: detection/detection_rotten_fruits diff --git a/config/models/detection/yolov5fruits_pretrain.yaml b/config/models/detection/yolov5fruits_pretrain.yaml index bec4077b..87692d05 100644 --- a/config/models/detection/yolov5fruits_pretrain.yaml +++ b/config/models/detection/yolov5fruits_pretrain.yaml @@ -1,4 +1,4 @@ _target_: ultralytics.YOLO description: yolov5 by ultralytics -model: /home/qazybek/restart/innofw/dui/logs/train/rotten_fruits/example_image_detection_yolo.yaml/20240826-203849/checkpoints/best.pt +model: yolov5xu name: yolov5 \ No newline at end of file