Skip to content

Commit 30b16c8

Browse files
author
yichen
committed
update keras
1 parent 76a0d48 commit 30b16c8

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/ml/basic/keras_mnist.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
#!/usr/bin/env python
22
# -*- coding: utf-8 -*-
33

4+
# you should write CUDA_VISIBLE_DEVICES=0 python test.py if want run with CPU
45
from __future__ import print_function
56
import keras
67
from keras.datasets import mnist

0 commit comments

Comments
 (0)