在pytorch kmeans中使用pytorch Dataloader

问题描述

我正在尝试使用Pytorch-kmeans(github)执行Kmeans聚类。我有大约27M数组,每个数组约512个元素。 numpy数组的大小为51GB,大于我的GPU RAM(32GB)。有什么方法可以使用pytorch DataLoader类批处理数组吗?

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)