Merge branch 'master' into docs

aimclub · Oct 6, 2023 · 49f156f · 49f156f
2 parents 23db22b + 1cb7a9d
commit 49f156f
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/.github/workflows/pep8-test.yaml b/.github/workflows/pep8-test.yaml
@@ -21,4 +21,4 @@ jobs:
         pip install flake8
     - name: Run PEP8 test
       run: |
-        flake8 --max-line-length=130 --exclude=ODRS/train_utils/train_model/models/*,ODRS/ml_utils/*,ODRS/train_utils/custom_train_all.py,ODRS/train_utils/train_model/scripts/yolov7_train.py,tests/* .
+        flake8 --max-line-length=130 --exclude=ODRS/train_utils/train_model/models/*,ODRS/ml_utils/*,ODRS/train_utils/custom_train_all.py,ODRS/train_utils/train_model/scripts/yolov7_train.py,tests/*,docs/* .
diff --git a/ODRS/train_utils/config/custom_config.yaml b/ODRS/train_utils/config/custom_config.yaml
@@ -2,7 +2,7 @@ BATCH_SIZE: 10
 CLASSES: classes.txt
 CONFIG_PATH: dataset.yaml
 DATA_PATH: /home/runner/work/ODRS/ODRS/user_datasets/yolo/aeral
-EPOCHS: 10
+EPOCHS: 2
 GPU_COUNT: 0
 IMG_SIZE: 300
 MODEL: yolov5m

diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -1,3 +1,4 @@
 sphinx
 sphinx_rtd_theme
-autodocsumm
+autodocsumm
+