search

ZhangYuanhan-AI / CelebA-Spoof

[ECCV2020] A Large-Scale Face Anti-Spoofing Dataset
531 stars 92 forks source link

RuntimeError: CUDA out of memory. #34

Closed HsiaoCH closed 3 years ago

HsiaoCH commented 3 years ago

Screenshot from 2020-11-30 15-59-03 我在執行main.py時遇到此問題，關於此我應該如何解決？

ZhangYuanhan-AI commented 3 years ago

Reset the batch size.

HsiaoCH commented 3 years ago

batch size 會對結果造成什摸影響？ 目前我改為256

ZhangYuanhan-AI commented 3 years ago

From my experience. With a larger batch size,

  1. Multi-step is not essential.
  2. Performance can obviously improve.
  3. Model converge will speed up.
HsiaoCH commented 3 years ago

Thanks reply