torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

remember split_embedding_weights

Open YLGH opened this issue 3 years ago • 1 comments

Summary: ATT currently calling split_embedding_weights repeatedly over FBGEMM kernels provide tensor views that are identical, but have different ids (hash values). As a result, it makes comparing tensors from state_dict tricky

Differential Revision: D37337616

YLGH avatar Jun 22 '22 05:06 YLGH

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

facebook-github-bot avatar Jun 22 '22 05:06 facebook-github-bot