Move state dict to cpu before converting to state stream

ray-project / ray_lightning

Pytorch Lightning Distributed Accelerators using Ray

Apache License 2.0

211 stars 34 forks source link

Closed MarkusSpanring closed 1 year ago

MarkusSpanring commented 1 year ago

PR related to #207

cc @amogkam

MarkusSpanring commented 1 year ago

Can this be merged, since all checks have passed, or is something else required from my side?