PH8411
PH8411
可能是你自己数据集和coco数据集的差距造成的，导致一开始描述子差距很大，预测的得分较小；superglue_train.py文件里损失函数代码：loss.append(-torch.log( scores[0][x][y].exp() ))改为loss.append(-torch.log( scores[0][x][y].exp() )+epison),其中epison是一个小量，设置为1e-6，可以避免得分小的时候，loss无穷大。 如果不是这个原因，试试降低学习率
> @shahizat have you successed run TensorRT-LLM on the Nvidia AGX Orin development kit ? same question