We will gradually release our tools for profiling performance and memory consumption of DNN training here. For now, we released our tool for profiling the memory consumption based on MXNet.
marcoszh / tbd-tools Goto Github PK
View Code? Open in Web Editor NEWThis project forked from tbd-ai/tbd-tools