levelcapgg
Results
2
comments of
levelcapgg
> Hi, @Riga27527 , here is my solution: > > ``` > def move_chunks_to_cpu(chunks): > chunks_new = [] > device_cpu = jax.devices("cpu")[0] > for chunk in chunks: > chunk_new =...
> > > Hi, @Riga27527 , here is my solution: > > > ``` > > > def move_chunks_to_cpu(chunks): > > > chunks_new = [] > > > device_cpu =...