CUDA out of memory
nicolasNi opened this issue · comments
随着不断的训练显存占用逐渐升高，最终会超出显存。
你好，使用32GB的显卡应该不会超出最大显存，如果是使用显存较小的显卡，在数据处理的时候可以适当降低最大宽高。
如果是降低最大宽高，那么会影响最终的ExpRate吗？
如果是降低最大宽高，那么会影响最终的ExpRate吗？
训练的时候数据是随机取的，如果一个batch里面既有很高的图像又有很宽的图像，那么这个batch的宽高会比较大，可以设置条件进行过滤，过滤掉的图像只是不在这个batch内部出现，所以应该不太会影响。
我想我应该明白了，谢谢你的回复
你好，使用32GB的显卡应该不会超出最大显存，如果是使用显存较小的显卡，在数据处理的时候可以适当降低最大宽高。
你好，请问下，再哪里设置图像的最大宽高呢？我看config.yaml里面好像没有图像宽高相关的配置