michael/modrec-workflow
1
0
Fork 0
forked from qoherent/modrec-workflow
Code Pull Requests Workflows Activity

fixed paths

Browse Source
This commit is contained in:
Liyu Xiao 2025-06-16 09:39:13 -04:00
parent ce20909ebe
commit 1bf85251d9
2 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.riahub/workflows/workflow.yaml
View File

@ -76,7 +76,7 @@ jobs:
uses: actions/upload-artifact@v3 uses: actions/upload-artifact@v3
with: with:
name: checkpoints name: checkpoints
path: checkpoint_files/inference_recognition_model.ckpt path: checkpoint_files/*
- name: 4. Convert to ONNX file - name: 4. Convert to ONNX file
run: | run: |

4
scripts/training/train.py
View File

@ -1,7 +1,7 @@
import sys, os import sys, os
os.environ["NNPACK"] = "0" os.environ["NNPACK"] = "0"
import lightning as L import lightning as L
from lightning.pytorch.callbacks import ModelCheckpoint, Callback from lightning.pytorch.callbacks import ModelCheckpoint
import torch import torch
import torch.nn.functional as F import torch.nn.functional as F
import torchmetrics import torchmetrics
@ -23,8 +23,6 @@ class CustomProgressBar(TQDMProgressBar):
def train_model(): def train_model():
settings = get_app_settings() settings = get_app_settings()
training_cfg = settings.training training_cfg = settings.training
dataset_cfg = settings.dataset
train_flag = True train_flag = True
batch_size = training_cfg.batch_size batch_size = training_cfg.batch_size
epochs = training_cfg.epochs epochs = training_cfg.epochs