We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 76a0d48 commit 30b16c8Copy full SHA for 30b16c8
src/ml/basic/keras_mnist.py
@@ -1,6 +1,7 @@
1
#!/usr/bin/env python
2
# -*- coding: utf-8 -*-
3
4
+# you should write CUDA_VISIBLE_DEVICES=0 python test.py if want run with CPU
5
from __future__ import print_function
6
import keras
7
from keras.datasets import mnist
0 commit comments