disabled gpu for now

.riahub/workflows/workflow.yaml

@@ -46,6 +46,7 @@ jobs:
 
       - name: 2. Train Model
         run: |
+          mkdir -p data/dataset
           PYTHONPATH=. python data/training/train.py
           echo "training model"
 

conf/app.yaml

@@ -21,7 +21,7 @@ training:
 inference:
   model_path: checkpoints/inference_recognition_model.ckpt
   num_classes: 4
-  output_path: results/inference_recognition_model.onnx
+  output_path: onnx_files/inference_recognition_model.onnx
 
 app:
   build_dir: dist

data/training/train.py

@@ -141,7 +141,7 @@ def train_model():
     trainer = L.Trainer(
         max_epochs=epochs,
         callbacks=[checkpoint_callback],
-        accelerator="gpu",
+        accelerator="cpu",
         devices=1,
         benchmark=True,
         precision="bf16-mixed",