benchmark icon indicating copy to clipboard operation
benchmark copied to clipboard
verified publisher icon pytorch

Rewrite DLRM to be TorchScript-able

Open gmagogsfm opened this issue 4 years ago • 0 comments

  • The only portion not scriptable is parallel_forward calling into nn.parallel, but it is only needed for distributed execution, which as far as I know isn't part of plan for benchmarking.

gmagogsfm avatar Mar 13 '21 07:03 gmagogsfm