Memory-efficient Model Weight Loading#
This folder contains code to illustrate how to load model weights more efficiently
memory-efficient-state-dict.ipynb: contains code to load model weights via PyTorch’s
load_state_dictmethod more efficiently
This folder contains code to illustrate how to load model weights more efficiently
memory-efficient-state-dict.ipynb: contains code to load model weights via PyTorch’s load_state_dict method more efficiently