Tracking memory consumption in python script during execution