Hang Su

I am not sure whether that matters but we only tested with Python3. Can you please report your Pytorch version? Also, do all the tests pass?

`python -m unittest` will run all tests. Seeing whether any fails and which ones can help us diagnose.

Sorry for the late reply. Have you solved it? If not, can you provide more context of the issue (hardware/system info, whether test cases pass, etc.)?

Sorry for the delay! There are ReLU layers after pac_t layers. See https://github.com/NVlabs/pacnet/blob/master/task_jointUpsampling/models.py#L248 The output of the last conv is the output of the network. Depending on the task, it...

We utilize im2col as part of the internal computation of our layer, and that requires relatively large memory allocation (but subsequently released). It can be helpful to profile your peak...

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