Priyesh Vijayan
Results
1
issues of
Priyesh Vijayan
How to restore a variable from checkpoint saved in cpu back in cpu when you have both gpu and cpu?
5
I get the following error when I try to restore, **ValueError: SingleDeviceSharding with Device=TFRT_CPU_0 was not found in jax.local_devices().** Despite enclosing the statements within a CPU device scope, like below,...