Module.cpp File Reference

#include "nnkit/support/caffe/Backend.h"
#include <nnkit/CmdlineArguments.h>
#include <memory>

Go to the source code of this file.

Functions
std::unique_ptr< nnkit::Backend >	make_backend (const nnkit::CmdlineArguments &args)

Function Documentation

make_backend()

std::unique_ptr< nnkit::Backend > make_backend

(

const nnkit::CmdlineArguments &

args

)

Definition at line 23 of file Module.cpp.

{
  using std::make_unique;
 
  auto net = make_unique<::caffe::Net<float>>(args.at(0), caffe::TEST);
 
  if (args.size() > 1)
  {
    net->CopyTrainedLayersFrom(args.at(1));
  }
 
  return make_unique<::nnkit::support::caffe::Backend<float>>(std::move(net));
}