Loss is nan, stopping training

kakaobrain / hotr

Official repository for HOTR: End-to-End Human-Object Interaction Detection with Transformers (CVPR'21, Oral Presentation)

Apache License 2.0

140 stars 19 forks source link

Closed widedh closed 1 year ago

widedh commented 1 year ago

I have an issue with the training, for the 2 datasets, I get the same error: Loss is nan, stopping training.

widedh commented 1 year ago

I just changed the parameters of training. It was mainly an issue related to my GPU.