[jira] [Created] (SINGA-548) Switch between CPU and GPU devices

[Luo Zhaojing (Jira)]

Luo Zhaojing created SINGA-548:
----------------------------------

             Summary: Switch between CPU and GPU devices
                 Key: SINGA-548
                 URL: https://issues.apache.org/jira/browse/SINGA-548
             Project: Singa
          Issue Type: Improvement
            Reporter: Luo Zhaojing


For cnn and cifar_distributed_cnn examples, in the train_cnn.py, the models can 
only be put in GPU device. This should be changed to support both GPU and CPU



--
This message was sent by Atlassian Jira
(v8.20.10#820010)