fix-model-test

aimclub · Feb 14, 2024 · bbfca95 · bbfca95
1 parent 1d15998
commit bbfca95
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 10 deletions.
diff --git a/.github/workflows/ml-test.yaml b/.github/workflows/ml-test.yaml
@@ -21,7 +21,6 @@ jobs:
         pip install -r requirements.txt
     - name: Download dataset
       run: |
-        pwd
         cd user_datasets
         git clone https://github.com/AIRI-Institute/WaRP
         mkdir WaRP/Warp-D/valid

diff --git a/.github/workflows/model-test.yaml b/.github/workflows/model-test.yaml
@@ -22,9 +22,9 @@ jobs:
     - name: Download dataset
       run: |
         cd user_datasets
-        mkdir aeral
-        cd aeral
-        curl -L "https://public.roboflow.com/ds/et6MNsOwMJ?key=GMSDIzpYnW" > roboflow.zip; unzip roboflow.zip; rm roboflow.zip
+        git clone https://github.com/AIRI-Institute/WaRP
+        mkdir WaRP/Warp-D/valid
+        cp -r WaRP/Warp-D/test/* WaRP/Warp-D/valid/
     - name: AI training
       run: |
         python ODRS/train_utils/custom_train_all.py
diff --git a/ODRS/train_utils/config/custom_config.yaml b/ODRS/train_utils/config/custom_config.yaml
@@ -1,11 +1,11 @@
-BATCH_SIZE: [25, 20]
+BATCH_SIZE: 20
 CLASSES: classes.txt
-DATA_PATH: Warp-D
-EPOCHS: [4, 5]
+DATA_PATH: /home/runner/work/ODRS/ODRS/user_datasets/WaRP/Warp-D
+EPOCHS: 2
 GPU_COUNT: 1
-IMG_SIZE: [300, 400]
-MODEL: [yolov5l, yolov5s]
-SELECT_GPU: 0
+IMG_SIZE: 300
+MODEL: yolov5s
+SELECT_GPU: cpu
 
 CONFIG_PATH: dataset.yaml
 SPLIT_TRAIN_VALUE: 0.6