torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Refactor passing over cache params

Open henrylhtsang opened this issue 8 months ago • 3 comments

Summary: Refactor the passing over cache params from dataclass to fused_params dict a bit.

Differential Revision: D58886177

henrylhtsang avatar Jun 21 '24 19:06 henrylhtsang