Can you add the feature of printing variable's name

[HaoranZhuExplorer]

God job in this project! Can you add one feature which can print variable's name?(Intermediate variables do not belong to network architectures. ). When I want to track how my gpu memory is used, the tool has clearly helped how many memories I've used between lines, I think printing variable names will be even helpful!

Best regards