When I try to export mesh using this script:
python utils/export.py checkpoints/a_high_quality_photo_of_a_corgi/2023-10-24/140914/ckpts/step_14000.pt --type mesh --batch_size 65536 --reso 256 --K 200 --thresh 0.1
I meet this error:
File "/mnt/sda//gsgen/utils/export.py", line 138, in to_mesh
density_val_grid, L = get_density_val_grid_from_ckpt(
File "/mnt/sda//gsgen/utils/export.py", line 94, in get_density_val_grid_from_ckpt
_, nn_idx, dist = K_nearest_neighbors(
File "/mnt/sda//anaconda3/envs/gsgen/lib/python3.9/site-packages/torch/autograd/grad_mode.py", line 27, in decorate_context
return func(*args, **kwargs)
File "/mnt/sda//gsgen/utils/ops.py", line 129, in K_nearest_neighbors
dist, idx, nn = knn_points(query[None, ...], mean[None, ...], K=K, return_nn=True)
File "/mnt/sda//anaconda3/envs/gsgen/lib/python3.9/site-packages/pytorch3d/ops/knn.py", line 187, in knn_points
p1_dists, p1_idx = _knn_points.apply(
File "/mnt/sda//anaconda3/envs/gsgen/lib/python3.9/site-packages/pytorch3d/ops/knn.py", line 72, in forward
idx, dists = _C.knn_points_idx(p1, p2, lengths1, lengths2, norm, K, version)
RuntimeError: p1 must be a CUDA tensor.
When I try to export mesh using this script:
python utils/export.py checkpoints/a_high_quality_photo_of_a_corgi/2023-10-24/140914/ckpts/step_14000.pt --type mesh --batch_size 65536 --reso 256 --K 200 --thresh 0.1
I meet this error: