Commits on Aug 22, 2024

1. Add an input debug function in TBE training (pytorch#3022) …

   Summary:
   X-link: facebookresearch/FBGEMM#117
   
   Pull Request resolved: pytorch#3022
   
   This diff adds a function for printing input stats including
   weighted/unweighted, number of features, batch size, average pooling
   factor, total number of indices, number of unique indices, and number
   of indices that goes through the different backward functions. This
   function is intended for debugging only. It can enabled during runtime
   by setting an enviroment variable FBGEMM_DEBUG_PRINT_INPUT_STATS=1.
   
   Reviewed By: dshi7
   
   Differential Revision: D61634512

   

   sryap authored and facebook-github-bot committed Aug 22, 2024
   Configuration menu

   • View commit details
   • Copy full SHA for 082d5c3
   • Browse repository at this point

   Copy the full SHA

   082d5c3 View commit details

   Browse the repository at this point in the history