search

skyhehe123 / SA-SSD

SA-SSD: Structure Aware Single-stage 3D Object Detection from Point Cloud (CVPR 2020)
492 stars 106 forks source link

TypeError: grid_sample() got an unexpected keyword argument 'align_corners' #74

Open praj441 opened 3 years ago

praj441 commented 3 years ago

While training for multi_class, I am getting this error. My PyTorch version is 1.1.0. I searched on google and find out that the attribute 'align_corners' is introduced in PyTorch 1.3.0. I request the author to update its Dependency requirement in ReadMe.

praj441 commented 3 years ago

The alternative and simple solution is to remove the argument 'align_corners=True' from the function "torch.nn.functional.grid_sample" in file mmdet/models/single_stage_heads/ssd_rotate_head.py But I don't know if this can cause performance issues.