I'm getting the following error when running on a system with 2gb nvidia geforce 940M card with ompute capability 5.0
ResourceExhaustedError (see above for traceback): OOM when allocating tensor with [shape[1,10,400,352,128]]
can you please tell me if i can run training with my current system configuration. If so, please give me the suitable input arguments with train.py so as to avoid over usage of gpu.
Hey Jeasinema!
I'm getting the following error when running on a system with 2gb nvidia geforce 940M card with ompute capability 5.0
ResourceExhaustedError (see above for traceback): OOM when allocating tensor with [shape[1,10,400,352,128]
] can you please tell me if i can run training with my current system configuration. If so, please give me the suitable input arguments with train.py so as to avoid over usage of gpu.