Regarding error when running (DCC.py)

shahsohil / DCC

This repository contains the source code and data for reproducing results of Deep Continuous Clustering paper

MIT License

209 stars 53 forks source link

Closed mz3220 closed 3 years ago

mz3220 commented 4 years ago

**Hello

Thank you for this project

When running (DCC.py), I face the following run-time error:

builtins.IndexError: tensors used as indices must be long, byte or bool tensors

Can you help me with it.

wangxinzhi0 commented 4 years ago

Please have a try like this, convert Tensor to long

    # print(type(pairs),type(index)) # Tensor
    long_index = index.long()
    centroids = self.U[long_index]