Open praj441 opened 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.
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.