PyTorch Forums
Can I use CPU mem to save GPU mem at sequential model?
nlp
ptrblck
June 14, 2022, 5:55am
2
You could check the
CPU offloading
for your use case.
show post in topic