Skip to content

预测模型权重加载 #1

@PDY123

Description

@PDY123

您好,想请教以下:
代码内提供的预训练FENGWU模型,只能在batchsize=1可以训练,虽然fengwu模型设置是动态batchsize,但依旧会报错,想请教一下是什么原因。
如果需要全局batchsize为16,只能通过修改worldsize吗?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions