diff --git a/.github/workflows/ml-test.yaml b/.github/workflows/ml-test.yaml
index b59a085..0434ed3 100644
--- 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
index 0f7698c..71624f8 100644
--- 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
\ No newline at end of file
diff --git a/ODRS/train_utils/config/custom_config.yaml b/ODRS/train_utils/config/custom_config.yaml
index 86f093c..ab48254 100755
--- 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