Skip to content

Latest commit

 

History

History
13 lines (10 loc) · 312 Bytes

File metadata and controls

13 lines (10 loc) · 312 Bytes

Basic MNIST Example

pip install -r requirements.txt

# For training the fully connnected net
python fcn.py
# CUDA_VISIBLE_DEVICES=2 python fcn.py  # to specify GPU id to ex. 2

# For training the convolution net
python conv.py
# CUDA_VISIBLE_DEVICES=2 python conv.py  # to specify GPU id to ex. 2