Closed Brandonnogithub closed 5 years ago
I run this model on 1080ti command: python train.py --dataset rocstories --desc rocstories --submit --analysis Then I got ResourceExhaustedError Do I need to make some changes to the code before running? Thanks.
I solve it. The default #GPU is 4. Just set it as 1.
I run this model on 1080ti command: python train.py --dataset rocstories --desc rocstories --submit --analysis Then I got ResourceExhaustedError Do I need to make some changes to the code before running? Thanks.