search

Haiyang-W / DSVT

[CVPR2023] Official Implementation of "DSVT: Dynamic Sparse Voxel Transformer with Rotated Sets"
https://arxiv.org/abs/2301.06051
Apache License 2.0
361 stars 28 forks source link

[Question] Has this been tested on Pytorch 2.0? #11

Closed MyronRodrigues-StreetDrone closed 1 year ago

MyronRodrigues-StreetDrone commented 1 year ago

Hey thanks for open sourcing your work.

Leaving this question here so people can see/answer if anyone tests this on Pytorch 2.0

chenshi3 commented 1 year ago

We do not test the code on Pytorch 2.0, and we use Pytorch 1.9.1

Haiyang-W commented 1 year ago

You can try to be compatible with PyTorch 2.0, which will make model training faster.

YoushaaMurhij commented 1 year ago

It works fine with Pytorch 2.0.0

Haiyang-W commented 1 year ago

Great!

MyronRodrigues-StreetDrone commented 1 year ago

thanks @YoushaaMurhij