08_memory_efficient_weight_loading

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_dict method more efficiently