liuruijin17 / LSTR

This is an official repository of End-to-end Lane Shape Prediction with Transformers.
BSD 3-Clause "New" or "Revised" License
644 stars 130 forks source link

Training stop #85

Open wohaiyo opened 2 years ago

wohaiyo commented 2 years ago

When training the model, the program often stops at an iteration without error information. At this time, the memory is occupied. The program also stops outputting. I am using one Nvidia RTX.