JasonChenJC

[BUG/Help] cannot import name 'skip_init' from 'torch.nn.utils'

15

### Is there an existing issue for this? - [X] I have searched the existing issues ### Current Behavior  torch=2.0.0 ### Expected Behavior _No response_ ### Steps To Reproduce...

In KMeans_v1.py, "if dim != 3" should be "if dim != 2"?