Priyesh Vijayan comments

Repositories
Issues
Comments

Results 2 comments of


                                            Priyesh Vijayan

How to restore a variable from checkpoint saved in cpu back in cpu when you have both gpu and cpu?

Hi, I don't have any sharding settings specified for the variable. Below is the entirety of the code. PS: I'm running on a machine with a GPU ``` import jax...

How to restore a variable from checkpoint saved in cpu back in cpu when you have both gpu and cpu?

I'm sorry for being unclear. I'll try to explain again, if you don't mind. Objective: Load a variable (originally in CPU) into CPU memory. Issue: When I call restore, it...