Skip to content

Instantly share code, notes, and snippets.

@aabadie aabadie/
Last active May 18, 2016

What would you like to do?
Base python script used to profile memory usage of joblib dump/load functions
import numpy as np
import joblib
obj = [np.ones((5000, 5000)), np.random.random((5000, 5000))]
joblib.dump(obj, '/tmp/test.pkl', compress=True)

This comment has been minimized.

Copy link
Owner Author

aabadie commented May 17, 2016

How to use this script

  1. Add @profile decorator to joblib dump and load functions in joblib/
  2. Run the following command in a console:
$ mprof run && mprof plot
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.