A tensor size mismatch bug when traing

bradyz / cross_view_transformers

Cross-view Transformers for real-time Map-view Semantic Segmentation (CVPR 2022 Oral)

MIT License

531 stars 81 forks source link

A tensor size mismatch bug when traing #17

Closed zjuliangxun closed 2 years ago

zjuliangxun commented 2 years ago

Hi, thanks for your outstanding work! When I was training this model with default configurations, a size mismatch error came up. This error is mainly related to the class "SigmoidFocalLoss", and happens just before the epoch 0. Is there any bug or I did something wrong? Screenshot from 2022-06-09 12-42-56

zjuliangxun commented 2 years ago

Sorry ,that's my fault