fengyuan1993012

the function torch.cdist cloud compute the distance. I have written the function def knntry(pc,n_neighbors,r,h): # dist = torch.cdist(pc,pc) distnew = -dist neigbhors = distnew.topk(k=n_neighbors+1,dim=2,largest=False) distvalue = neigbhors.values[:,:,1:] distvalue = -distvalue...

https://github.com/DLR-RM/stable-baselines3 my system is windows10, pytroch 2.2.0, stable-baselines3 is friendly