Skip to content

预测模型权重加载 #1

Description

@PDY123

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

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

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