search

pytorch / FBGEMM

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
Other
1.17k stars 474 forks source link

Add an input debug function in TBE training #3022

Closed sryap closed 3 weeks ago

sryap commented 3 weeks ago

Summary: 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.

Differential Revision: D61634512

netlify[bot] commented 3 weeks ago

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
Latest commit 082d5c3066f92f0a3a331bd9b797b60b2fb2e5a7
Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/66c6950d66671f00088bb6aa
Deploy Preview https://deploy-preview-3022--pytorch-fbgemm-docs.netlify.app
Preview on mobile
Toggle QR Code...

QR Code

Use your smartphone camera to open QR code link.

To edit notification comments on pull requests, go to your Netlify site configuration.

facebook-github-bot commented 3 weeks ago

This pull request was exported from Phabricator. Differential Revision: D61634512

facebook-github-bot commented 3 weeks ago

This pull request was exported from Phabricator. Differential Revision: D61634512

facebook-github-bot commented 3 weeks ago

This pull request was exported from Phabricator. Differential Revision: D61634512

facebook-github-bot commented 3 weeks ago

This pull request has been merged in pytorch/FBGEMM@0861d469127ef880b448eaed186a60e4480b1e88.