added in type casting

.riahub/workflows/workflow.yaml

@@ -37,12 +37,12 @@ jobs:
           python -m pip install --upgrade pip
           pip install -r requirements.txt
 
-      - name: 1. Build HDF5 Dataset
-        run: |
-         mkdir -p data/dataset
-         PYTHONPATH=. python data/scripts/produce_dataset.py
-         echo "datasets produced successfully"
-        shell: bash
+      # - name: 1. Build HDF5 Dataset
+      #   run: |
+      #    mkdir -p data/dataset
+      #    PYTHONPATH=. python data/scripts/produce_dataset.py
+      #    echo "datasets produced successfully"
+      #   shell: bash
       
       - name: Upload Dataset Artifacts
         uses: actions/upload-artifact@v3

data/training/train.py

@@ -85,7 +85,7 @@ def train_model():
     hparams = {
         "drop_path_rate": 0.2,
         "drop_rate": 0.5,
-        "learning_rate": training_cfg.learning_rate,
+        "learning_rate": float(training_cfg.learning_rate),
         "wd": 0.01,
     }