Skip to content

Conversation

@KunikaValecha
Copy link

This code can can run for small number of images on cpu. It has been tested successfully on pytorch version = 1.4.0+cpu

@skhadem
Copy link
Owner

skhadem commented May 16, 2021

Hi! Sorry for the delay in responding to this. I see that the only thing that has changed is removing the calls to .cuda(). What do you think about instead adding a flag (something like --cpu_only) directly to Run.py? Then, whenever there is a call to cuda(), if this new flag is set we can keep the variables on the spu. This way we avoid duplicated code. Let me know if you have any questions!

@KunikaValecha
Copy link
Author

Yes, that would work.

@skhadem
Copy link
Owner

skhadem commented May 24, 2021

Do you want to make the changes in your branch and push directly to this PR?

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.

2 participants