Skip to content

关于batch size对于模型测试结果的影响 #60

Description

@zhz-bit

你好,我发现当使用batchsize为16训练好模型之后,修改测试集的batchsize会出现指标不一样的问题。

  1. 当使用batchsize=16时,
    Image
  2. 相同的预训练权重,当batchsize=32时,
    Image

我发现,batchsize作为了数据集的一个param,
Image
但是dataloader是在从npz文件加载数据之后再分配的,所以batchsize这个参数会对已保存的数据集造成什么影响吗?
Image

也许对pad操作会影响最后一个batchsize的样本
Image
但是我们对两个预测数据进行输出,对于测试集的第一个样本,相同的真值,但是预测值在小数点后最后几位是有所不同的,请问您之前是否发现了这个问题呢?
Image

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