我的 GPU 是 Rtx 3050 4GB。由于 VRAM 较少,我减少了批量大小,但仍然花费太长的时间,每个时期几乎需要 1 小时。我可以保存训练好的模型(.h5)并重新训练它而不丢失任何进度吗?
我还没有尝试过任何东西,因为我是机器学习的新手。我正在使用tensorflow和keras
我还没有尝试过,我不确定这种方式是否有效。最好将图像调整为较小的尺寸,例如 (28,28)。这样,纪元会运行得更快,并且需要更少的 GPU 进行计算。