configured workflow files to run all 3 steps

.gitignore

@@ -2,3 +2,7 @@
 
 # Byte-compiled / optimized / DLL files
 __pycache__/
+*.h5
+*.ckpt
+*.ipynb
+*.onnx

.riahub/workflows/workflow.yaml

@@ -39,20 +39,20 @@ jobs:
 
       - name: 1. Build HDF5 Dataset
         run: |
-        #  mkdir -p data/dataset
+         mkdir -p data/dataset
          PYTHONPATH=. python data/scripts/produce_dataset.py
          echo "datasets produced successfully"
         shell: bash
 
       - name: 2. Train Model
         run: |
+          PYTHONPATH=. python data/training/train.py
           echo "training model"
-        # Placeholder
 
       - name: 3. Build inference app
         run: |
+          PYTHONPATH=. python convert_to_onnx.py
           echo "building inference app"
-        # Placeholder
 
       - name: Upload Dataset Artifacts
         uses: actions/upload-artifact@v3
@@ -60,15 +60,15 @@ jobs:
           name: ria-dataset
           path: data/dataset/**
 
-    #  - name: Upload Checkpoints
-    #    uses: actions/upload-artifact@v3
-    #    with:
-    #      name: ria-checkpoints
-    #      path: checkpoints/**
+      - name: Upload Checkpoints
+        uses: actions/upload-artifact@v3
+        with:
+          name: ria-checkpoints
+          path: checkpoint_files/inference_recognition_model.ckpt
 
 
-    #  - name: Upload Inference App
-    #    uses: actions/upload-artifact@v3
-    #    with:
-    #      name: ria-demo-app
-    #      path: dist/**
+      - name: Upload Inference App
+        uses: actions/upload-artifact@v3
+        with:
+          name: ria-demo-app
+          path: onnx_files/inference_recognition_model.onnx

conf/app.yaml

@@ -15,13 +15,13 @@ training:
   batch_size: 64
   epochs: 50
   learning_rate: 0.001
-  checkpoint_path: checkpoints/model.ckpt
+  checkpoint_path: checkpoints/inference_recognition_model.ckpt
   use_gpu: true
 
 inference:
-  model_path: checkpoints/model.ckpt
-  num_classes: 10
-  output_path: results/output.json
+  model_path: checkpoints/inference_recognition_model.ckpt
+  num_classes: 4
+  output_path: onnx_files/inference_recognition_model.onnx
 
 app:
   build_dir: dist

requirements.txt

@@ -1,4 +1,12 @@
-h5py
-numpy
-./wheel/utils-0.1.2.dev0-py3-none-any.whl
-
+h5py==3.13.0
+lightning==2.5.1.post0
+matplotlib==3.10.3
+numpy==2.2.6
+PyYAML==6.0.2
+PyYAML==6.0.2
+scikit_learn==1.6.1
+timm==1.0.15
+torch==2.7.0
+utils==1.0.2
+onnx==1.18.0
+./wheel/utils-0.1.2.dev0-py3-none-any.whl