qoherent/modrec-workflow
14
0
Fork 1
Code Issues 1 Pull Requests Workflows Packages Projects Releases 1 Wiki Activity

added in type casting
Some checks failed
RIA Hub Workflow Demo / ria-demo (push) Failing after 4m58s
Details

Browse Source
This commit is contained in:
Liyu Xiao 2025-05-26 10:06:18 -04:00
parent 489fecf113
commit b3d17f804c
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
.riahub/workflows/workflow.yaml
View File

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

2
data/training/train.py
View File

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