|
|
1
1
这里有一些事情可以尝试。 预处理输入
我不熟悉使用此参数
降维 如果图像尺寸大于224x224,则可以缩小图像尺寸。在许多情况下,你甚至可以试着缩小尺寸。 减少每个历元的步数 查看每个历元的步骤。根据上面的代码(patches.shape[0]),我不确定现在输入的是什么,但如果训练速度太慢,我不会比data\u size//batch\u size高出很多(你可以再高一些,因为它会增加数据,但需要更多的时间)。 升级设备 如果这些都不起作用,如果你还没有GPU,我建议你使用GPU。如果这是在CPU上完成的,那么这就是原因。对于CPU来说,图像变换是相对较大的矩阵计算。 |