自己采集的金属件图像比较大,image size : 2448*2048 ,每张图像上会有1 ~ 5 条直线 ,但是我自己训练出的效果很差,数据集大概 一百多张,batch size = 4 , gradient_accum_steps = 1 , 别的参数都没有修改。 我需要修改哪些参数才能提高训练效果呢? 另外 损失值 都是 nan 。
自己采集的金属件图像比较大,image size : 2448*2048 ,每张图像上会有1 ~ 5 条直线 ,但是我自己训练出的效果很差,数据集大概 一百多张,batch size = 4 , gradient_accum_steps = 1 , 别的参数都没有修改。 我需要修改哪些参数才能提高训练效果呢? 另外 损失值 都是 nan 。