IBM / pytorch-large-model-support

Large Model Support in PyTorch
Apache License 2.0
132 stars 19 forks source link

[Advice] You guys should put a parameter to Readme file: how many times this package extends for poor GPU memory #11

Open yingshaoxo opened 1 year ago

yingshaoxo commented 1 year ago

For example, for a 2G GPU, after using this package, it could be used to train some models that requires 4G GPU.


By the way, I really like to know the increasing ratio this package provided for GPU memory.