Doesn't work in colab

[fader111]

  4 # Sphere mesh in pytorch3d

----> 5 sphere_mesh = ico_sphere(4, device)
6
7 verts_rgb = torch.ones_like(sphere_mesh.verts_list()[0])[None]

6 frames
/usr/local/lib/python3.7/dist-packages/pytorch3d/ops/subdivide_meshes.py in subdivide_heterogenerous(self, meshes, feats)
290 # Add one new vertex at the midpoint of each edge by taking the average
291 # of the verts that form each edge.
--> 292 new_verts = verts[edges].mean(dim=1)
293 new_verts = torch.cat([verts, new_verts], dim=0)
294

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