search

Tsinghua-MARS-Lab / ViP3D

MIT License
133 stars 19 forks source link

[bug] TypeError: can't convert np.ndarray of type numpy.object_ #8

Open HankKung opened 1 year ago

HankKung commented 1 year ago

Hi I'm encountering the following error when training vip3d with exactly the same scripts and configs:

File "/media/user/data/ViP3D/plugin/vip3d/models/attention_dert3d.py", line 440, in feature_sampling lidar2img = reference_points.newtensor(lidar2img) # (B, N, 4, 4) TypeError: can't convert np.ndarray of type numpy.object. The only supported types are: float64, float32, float16, complex64, complex128, int64, int32, int16, int8, uint8, and bool.

Does anyone have idea how to fix this bug?

Many thanks!!!

GentleSmile commented 1 year ago

Have you modified the batch_size or something else?

HankKung commented 1 year ago

My bad I modified the fix_feats=True, samples_per_gpu=2 in /vip3d_resnet50_3frame.py I run it with 4 GPUs and the batch_size remains 8: bash tools/dist_train.sh plugin/vip3d/configs/vip3d_resnet50_3frame.py 4 --work-dir=vip3d_resnet50_3frame.1

GentleSmile commented 1 year ago

For larger batch_size, you can refer to this issue.