Skip to content

训练自己采集的金属件图像数据 #32

Description

@BeiHaigogogo

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions