From b3d17f804ce01215f8243dfa89fe5147ef514df9 Mon Sep 17 00:00:00 2001 From: Liyu Xiao Date: Mon, 26 May 2025 10:06:18 -0400 Subject: [PATCH] added in type casting --- .riahub/workflows/workflow.yaml | 12 ++++++------ data/training/train.py | 2 +- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/.riahub/workflows/workflow.yaml b/.riahub/workflows/workflow.yaml index 2304f4f..e1cac93 100644 --- a/.riahub/workflows/workflow.yaml +++ b/.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 diff --git a/data/training/train.py b/data/training/train.py index f2106ab..4a2a3b6 100644 --- a/data/training/train.py +++ b/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, }