Corrected ckpt paths

InnopolisUni · Aug 27, 2024 · c74c8df · c74c8df
1 parent ceda919
commit c74c8df
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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