Skip to content

Fix gpu#63

Open
SarahAlidoost wants to merge 10 commits into
mainfrom
fix_gpu
Open

Fix gpu#63
SarahAlidoost wants to merge 10 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
  • I ran the notebook on levante with gpu.

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

@SarahAlidoost SarahAlidoost mentioned this pull request Jul 1, 2026
@SarahAlidoost SarahAlidoost marked this pull request as ready for review July 3, 2026 14:00
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