Contiguous nn.Parameter

tensorly / torch

TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch

http://tensorly.org/torch/

BSD 3-Clause "New" or "Revised" License

70 stars 18 forks source link

Closed JeremieMelo closed 2 years ago

JeremieMelo commented 2 years ago

Make all nn.Parameter contiguous for apex gradClipper

JeanKossaifi commented 2 years ago

That's great, thank you @JeremieMelo, merging!