Daoming Chen

Results 2 comments of Daoming Chen

Here is what I do to make sure all tensors and their indexes are on the same device. add after mesh.py line 27 ```py self.v_mask = np.ones(len(self.vs), dtype=bool) self.v_mask =...

Maybe you're looking for ViViT https://github.com/rishikksh20/ViViT-pytorch?