eminorhan/del_checkpoint.py

Last active January 13, 2022 16:40

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/eminorhan/f66f92daa71e424d06f9fb050e364152.js"></script>
Save eminorhan/f66f92daa71e424d06f9fb050e364152 to your computer and use it in GitHub Desktop.

Download ZIP

delete large memory checkpoint after loading in pytorch

Raw

del_checkpoint.py

	checkpoint = torch.load(CHECKPOINT_PATH)
	model.module.load_state_dict(checkpoint['model_state_dict'])
	del checkpoint
	torch.cuda.empty_cache()

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment