Skip to content

Fix gpu#63

Draft
SarahAlidoost wants to merge 38 commits into
mainfrom
fix_gpu
Draft

Fix gpu#63
SarahAlidoost wants to merge 38 commits into
mainfrom
fix_gpu

Conversation

@SarahAlidoost

@SarahAlidoost SarahAlidoost commented Jul 1, 2026

Copy link
Copy Markdown
Member

closes #30

In this PR, I tested the device using meta and fake tensor, see test_train.py

🔴 this branch is base on #54 , the notebook should be updated later, and this should be merged after merging #54.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Add support GPU in the model

1 participant