updated workflow, deleted old recordings, using gpu for training

This commit is contained in:
Liyu Xiao 2025-05-22 15:57:40 -04:00
parent 44507493a3
commit 39f3116d93
30 changed files with 2 additions and 86 deletions

View File

@ -10,7 +10,7 @@ on:
jobs:
ria-demo:
runs-on: ubuntu-latest
runs-on: ubuntu-latest-2080
steps:
- name: Print GPU information

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

View File

@ -143,7 +143,7 @@ def train_model():
trainer = L.Trainer(
max_epochs=epochs,
callbacks=[checkpoint_callback],
accelerator="cpu",
accelerator="gpu",
devices=1,
benchmark=True,
precision="bf16-mixed",