gpt4 book ai didi

keras - 如何在训练期间更改批量大小?

转载 作者:行者123 更新时间:2023-12-04 14:22:48 25 4
gpt4 key购买 nike

在训练期间,在每个时期,我想更改批量大小(用于 experimental 目的)。
创建自定义 Callback似乎合适,但 batch_size不是 Model 的成员类(class)。

我看到的唯一方法是覆盖 fit_loop 并曝光 batch_size到每个循环的回调。在不使用回调的情况下,是否有更清洁或更快的方法?

最佳答案

对于在这里登陆的其他人,我发现在 Keras 中进行批量大小调整的最简单方法就是多次调用 fit(使用不同的批量大小):

model.fit(X_train, y_train, batch_size=32, epochs=20)

# ...continue training with a larger batch size
model.fit(X_train, y_train, batch_size=512, epochs=10)

关于keras - 如何在训练期间更改批量大小?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50606995/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com