Skip to content
Snippets Groups Projects
Normal view Permalink
NeuralNetwork.cpp 130 B
Newer Older
  • Learn to ignore specific revisions
    • Franck Dary's avatar
    Added base class to neural network
    Franck Dary committed
    1 2 
    #include "NeuralNetwork.hpp"
    Franck Dary's avatar
    Detect device (cpu or cuda) and use cuda if possible  
    Franck Dary committed
    3 4 
    torch::Device NeuralNetworkImpl::device(torch::cuda::is_available() ? torch::kCUDA : torch::kCPU);