A convenient function to run everything on the GPU.
(self, gpu_id=0, use_cudnn=False)
| 2220 | Net._RunAllOnGPU(arg.n, gpu_id, use_cudnn) |
| 2221 | |
| 2222 | def RunAllOnGPU(self, gpu_id=0, use_cudnn=False): |
| 2223 | """A convenient function to run everything on the GPU.""" |
| 2224 | self._RunAllOnGPU(self._net, gpu_id, use_cudnn) |
| 2225 | |
| 2226 | |
| 2227 |