rectified-flow-pytorch icon indicating copy to clipboard operation
rectified-flow-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon lucidrains

v0.2.0 5x decreased performance

Open erikpro008 opened this issue 10 months ago • 5 comments

I have tried the new version training on device "mps" Macbook and it is 5x slower than the previous version. The GPU does not seem to utilize 100% because normally I would hear fan noises, so there might be a bottleneck with the 'hyper_connections'.

Happy new year ^^.

erikpro008 avatar Jan 05 '25 03:01 erikpro008