pacnet icon indicating copy to clipboard operation
pacnet copied to clipboard

Published 20 hours ago • verified publisher icon NVlabs

PacConv3d and AMP

Open saruarlive opened this issue 4 years ago • 2 comments

Great work, Han. Does it support PacConv3d and PacTransposeConv3D? Does it support PyTorch native automatic mixed precision (AMP)? What will the computational cost if I implement it?

saruarlive avatar Jan 12 '21 19:01 saruarlive

3D variants were not implemented, and we do not have any plans to bring additions or improvements to this repo. We didn't test with AMP (only available since pytorch 1.6), but I can't think of reasons why the operators won't work with AMP (you might want to try our th14 branch).

suhangpro avatar Jan 12 '21 19:01 suhangpro

@suhangpro tried PacConv2d with torch 1.11 using the code from th14 branch and I got errors during the backward method. Created a new issue for the same. Please take a look

jsetty avatar Dec 04 '21 02:12 jsetty