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
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