search

subhadarship / kmeans_pytorch

kmeans using PyTorch
https://subhadarship.github.io/kmeans_pytorch
MIT License
470 stars 75 forks source link

Does it work on DDP mode of PyTorch? #40

Open Hugo-cell111 opened 1 year ago

Hugo-cell111 commented 1 year ago

Hi! I wonder if I can use the function on multiple GPUs and if it works on DDP mode of PyTorch? Thanks!