search

NVlabs / pacnet

Pixel-Adaptive Convolutional Neural Networks (CVPR '19)
https://suhangpro.github.io/pac/
Other
514 stars 79 forks source link

PacConv3d and AMP #24

Open saruarlive opened 3 years ago

saruarlive commented 3 years ago

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?

suhangpro commented 3 years ago

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).

jsetty commented 2 years ago

@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