Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Running SATURN with CPU #81

Open
cozetto opened this issue Dec 13, 2024 · 1 comment
Open

Running SATURN with CPU #81

cozetto opened this issue Dec 13, 2024 · 1 comment

Comments

@cozetto
Copy link

cozetto commented Dec 13, 2024

Dear Authors,

I understand that SATURN was made for GPU. However, with a server that operates only with CPU, I am attempting to run it on a CPU by modifying the command as follows:

!python ../../train-saturn.py
...
--device_num=-1

I am reaching out to confirm whether it is feasible to run the code on a CPU. Currently, the code is running without issues during the pretraining step, but taking significantly longer. My main concern is whether using a CPU instead of a GPU could cause problems in later stages of the integration process.

image

Thank you for your guidance!

@bitcometz
Copy link

I also want to run SATURN with CPU !!!!

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

No branches or pull requests

2 participants