fix_CUDNN_STATUS_NOT_SUPPORTED

gyshgx868 / graph-ter

The official PyTorch implementation of the CVPR2020 paper "GraphTER: Unsupervised Learning of Graph Transformation Equivariant Representations via Auto-Encoding Node-wise Transformations".

MIT License

57 stars 18 forks source link

fix_CUDNN_STATUS_NOT_SUPPORTED #6

Closed RicardoLanJ closed 3 years ago

RicardoLanJ commented 3 years ago

by adding .contiguous() after all permute function, pytorch version no longer limited to <= 1.2.0.