我正在使用 CNN(序列模型)进行眼睛检测。我可以保存训练好的模型并重新训练它而不丢失我的进度吗

问题描述 投票:0回答:1

我的 GPU 是 Rtx 3050 4GB。由于 VRAM 较少,我减少了批量大小,但仍然花费太长的时间,每个时期几乎需要 1 小时。我可以保存训练好的模型(.h5)并重新训练它而不丢失任何进度吗?

我还没有尝试过任何东西,因为我是机器学习的新手。我正在使用tensorflow和keras

python tensorflow keras epoch
1个回答
0
投票

我还没有尝试过,我不确定这种方式是否有效。最好将图像调整为较小的尺寸,例如 (28,28)。这样,纪元会运行得更快,并且需要更少的 GPU 进行计算。

© www.soinside.com 2019 - 2024. All rights reserved.