Add min peak memory as a theano config variables
use THEANOFLAGS=profiling.min_peak_memory=True to set it on