lar_dl_content::LArDLHelper Class Reference

LArDLHelper class. More...

#include "LArDLHelper.h"

Public Types
typedef torch::jit::script::Module	TorchModel
typedef torch::Tensor	TorchInput
typedef std::vector< torch::jit::IValue >	TorchInputVector
typedef at::Tensor	TorchOutput

Static Public Member Functions
static pandora::StatusCode	LoadModel (const std::string &filename, TorchModel &model)
	Loads a deep learning model. More...
static void	InitialiseInput (const at::IntArrayRef dimensions, TorchInput &tensor)
	Create a torch input tensor. More...
static void	Forward (TorchModel &model, const TorchInputVector &input, TorchOutput &output)
	Run a deep learning model. More...

Detailed Description

LArDLHelper class.

Definition at line 22 of file LArDLHelper.h.

Member Typedef Documentation

typedef torch::Tensor lar_dl_content::LArDLHelper::TorchInput

Definition at line 26 of file LArDLHelper.h.

typedef std::vector<torch::jit::IValue> lar_dl_content::LArDLHelper::TorchInputVector

Definition at line 27 of file LArDLHelper.h.

typedef torch::jit::script::Module lar_dl_content::LArDLHelper::TorchModel

Definition at line 25 of file LArDLHelper.h.

typedef at::Tensor lar_dl_content::LArDLHelper::TorchOutput

Definition at line 28 of file LArDLHelper.h.

Member Function Documentation

void lar_dl_content::LArDLHelper::Forward ( TorchModel &  model, const TorchInputVector &  input, TorchOutput &  output  )

static

Run a deep learning model.

Parameters

model	the model to run
input	the input to run over
output	the tensor to store the output in

Definition at line 41 of file LArDLHelper.cc.

Referenced by lar_dl_content::DlHitTrackShowerIdAlgorithm::Infer(), and lar_dl_content::DlVertexingAlgorithm::Infer().

{
    output = model.forward(input).toTensor();
}

void lar_dl_content::LArDLHelper::InitialiseInput ( const at::IntArrayRef  dimensions, TorchInput &  tensor  )

static

Create a torch input tensor.

Parameters

dimensions	the size of each dimension of the tensor: pass as {a, b, c, d} for example
tensor	the tensor to be initialised

Definition at line 34 of file LArDLHelper.cc.

Referenced by lar_dl_content::DlHitTrackShowerIdAlgorithm::Infer(), and lar_dl_content::DlVertexingAlgorithm::MakeNetworkInputFromHits().

{
    tensor = torch::zeros(dimensions);
}

StatusCode lar_dl_content::LArDLHelper::LoadModel ( const std::string &  filename, LArDLHelper::TorchModel &  model  )

static

Loads a deep learning model.

Parameters

filename	the filename of the model to load
model	the TorchModel in which to store the loaded model

Returns

STATUS_CODE_SUCCESS upon successful loading of the model. STATUS_CODE_FAILURE otherwise.

Definition at line 16 of file LArDLHelper.cc.

References e.

Referenced by lar_dl_content::DlHitTrackShowerIdAlgorithm::ReadSettings(), and lar_dl_content::DlVertexingAlgorithm::ReadSettings().

{
    try
    {
        model = torch::jit::load(filename);
        std::cout << "Loaded the TorchScript model \'" << filename << "\'" << std::endl;
    }
    catch (const std::exception &e)
    {
        std::cout << "Error loading the TorchScript model \'" << filename << "\':\n" << e.what() << std::endl;
        return STATUS_CODE_FAILURE;
    }

    return STATUS_CODE_SUCCESS;
}

---

The documentation for this class was generated from the following files:

• larpandoracontent/v04_08_01/source/larpandoradlcontent/LArHelpers/LArDLHelper.h
• larpandoracontent/v04_08_01/source/larpandoradlcontent/LArHelpers/LArDLHelper.cc