README.md in ruby-dnn-1.1.6 vs README.md in ruby-dnn-1.2.0

- old
+ new

@@ -93,11 +93,11 @@ | Pooling | MaxPool2D, AvgPool2D, GlobalAvgPool2D, UnPool2D | | Optimizers | SGD, Nesterov, AdaGrad, RMSProp, AdaDelta, RMSPropGraves, Adam, AdaBound | | Losses | MeanSquaredError, MeanAbsoluteError, Hinge, HuberLoss, SoftmaxCrossEntropy, SigmoidCrossEntropy | ## Datasets -By setting the environment variable "RUBY_DNN_DOWNLOADS_PATH", you can specify the path to read dataset. +By setting the environment variable 'RUBY_DNN_DOWNLOADS_PATH', you can specify the path to read dataset. ● Iris ● MNIST ● Fashion-MNIST ● CIFAR-10 @@ -107,13 +107,17 @@ ## Examples ● VAE ● DCGAN ● Pix2pix +## Use GPU +If you do 'require "cumo/narray"' before 'require "dnn"', you can run it on GPU. +Or, set the environment variable 'RUBY_DNN_USE_CUMO 'to'ENABLE' to force the GPU to be used. + ## TODO ● Write a test. ● Write a document. -● Support to GPU. +● Improve performance when using GPU. ## Development After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake "spec"` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.