Hi, @ethnhe
This is really nice work.
However, there seems to be a bug in meanshift_pytorch.py--》class MeanShiftTorch--》func fit
it seems that the "while True" cycling does not work. It seems that if I delete this cycle, and replace it with "Ar = A.view(1, N, c).repeat(N, 1, 1)", the result will not be changed.
Hi, @ethnhe This is really nice work. However, there seems to be a bug in meanshift_pytorch.py--》class MeanShiftTorch--》func fit it seems that the "while True" cycling does not work. It seems that if I delete this cycle, and replace it with "Ar = A.view(1, N, c).repeat(N, 1, 1)", the result will not be changed.