maybe some error in train_iq.py?

In line 186 of train_iq.py, perhaps "scheduler" should be changed to "info_scheduler"? Otherwise, it seems that "info_scheduler" has not been updated continuously? This may be the reason why the loss gradually becomes NaN during the training process.