【发布时间】:2021-12-28 23:13:17
【问题描述】:
显然,我知道添加验证数据会使训练花费更长的时间,但我在这里所说的时间差是荒谬的。代码:
# Training
def training(self, callback_bool):
if callback_bool:
callback_list = []
else:
callback_list = []
self.history = self.model.fit(self.x_train, self.y_train, validation_data=(self.x_test, self.y_test),
batch_size=1, steps_per_epoch=10, epochs=100)
尽管我的测试数据的大小是 10,000 个数据点,但上面的代码需要 30 多分钟来训练。我的训练数据大小为 40,000 个数据点,当我在没有验证数据的情况下进行训练时,我会在几秒钟内完成。有没有办法解决这个问题?为什么需要这么长时间?首先,我也在使用 gpu 进行训练!
【问题讨论】:
标签: python tensorflow machine-learning keras scikit-learn