Qtd de memoria de uma função - python

Pessoal, qual módulo do python eu posso usar para calcular a quantidade de memória gasta na execução de uma função? Eu to usando o memory_profiller, mas ele mostra a quantidade de memória gasta por cada linha do algoritmo, no caso eu quero um que mostre a quantidade total gasta.