pytorch model summary, statistic parameters number, memory usage, FLOPs and so on - View it on GitHub