Freewire icon indicating copy to clipboard operation
Freewire copied to clipboard

Published 20 hours ago • verified publisher icon noahtren

scatter and gather operations in Model forward pass

Open noahtren opened this issue 6 years ago • 0 comments

PyTorch 1.2 indexing isn't 100% as fast as scatter and gather operations. There might be a speed boost in using these operations instead of indexing in the Model class's forward method.

noahtren avatar Sep 14 '19 18:09 noahtren