tsunghan-wu / RandLA-Net-pytorch

:four_leaf_clover: Pytorch Implementation of RandLA-Net (https://arxiv.org/abs/1911.11236)
MIT License
122 stars 33 forks source link

Maybe shoule disabling cuDNN #11

Closed huixiancheng closed 2 years ago

huixiancheng commented 3 years ago

Experiments on torch1.60 with 2080ti show that disabling cuDNN results in a significant training speedup, however it is uncertain whether this affects performance. However, it's very strange and no idea about why cause it. Top is set torch.backends.cudnn.enabled = False. Under is default Settings. QQ截图20211007200226 .

QQ截图20211007202631