search

facebookresearch / 3detr

Code & Models for 3DETR - an End-to-end transformer model for 3D object detection
Apache License 2.0
629 stars 79 forks source link

Fixes for Inplace Errors When Running with Pytorch 1.10 #18

Closed stanleyshly closed 3 years ago

stanleyshly commented 3 years ago

Even though this repo does not support Pytorch 1.1.0 officially, I modified the code to work with Pytorch 1.10 by changing some operations to inplace=False. It does increase memory usage a tad bit though, around 2400 MB to 2700 MB, so ideally if a better solution exists, it would be better, but this does work for now.

stanleyshly commented 3 years ago

Training Performance around 2x as slow, needs further modifications

ruizhaoz commented 2 years ago

Even though this repo does not support Pytorch 1.1.0 officially, I modified the code to work with Pytorch 1.10 by changing some operations to inplace=False. It does increase memory usage a tad bit though, around 2400 MB to 2700 MB, so ideally if a better solution exists, it would be better, but this does work for now.

Could you tell me where you change about the inplace operation? I met the same bug.