Closed PolicyWsy closed 3 years ago
Hi @PolicyWsy,
I have not tried using Colab with this project. However, based on the snippet you should first import softpool_cuda
and then the PyTorch classes and functions.
Best, Alex
Hi @alexandrosstergiou,
Thanks for your reply. I tried to import softpool_cuda
first but ended up with the same error. Anyway, I rewrote softpool by myself and it works well;)
能否分享一下重写以后的softpool
Hi, I installed the package on Colab with the commands you provide. The installation process seemed successful, but when I tried to import
soft_pool2d
, I got the following error message:Is there any way to use SoftPool on Colab? Thanks in advance.