Rcs::TorchPolicy Class Reference

#include <TorchPolicy.h>

Inheritance diagram for Rcs::TorchPolicy:

Collaboration diagram for Rcs::TorchPolicy:

Public Member Functions
	TorchPolicy (const char *filename)
virtual	~TorchPolicy ()
virtual void	reset ()
virtual void	computeAction (MatNd *action, const MatNd *observation)
Public Member Functions inherited from Rcs::ControlPolicy
	ControlPolicy ()
virtual	~ControlPolicy ()

Private Attributes
torch::jit::script::Module	module

Additional Inherited Members
Public Types inherited from Rcs::ControlPolicy
typedef ControlPolicy *(*	ControlPolicyCreateFunction) (const char *)
Static Public Member Functions inherited from Rcs::ControlPolicy
static void	registerType (const char *name, ControlPolicyCreateFunction creator)
static ControlPolicy *	create (const char *name, const char *dataFile)
static ControlPolicy *	create (PropertySource *config)
static std::vector< std::string >	getTypeNames ()

Detailed Description

ControlPolicy backed by a scripted torch module.

Definition at line 44 of file TorchPolicy.h.

Constructor & Destructor Documentation

TorchPolicy()

Rcs::TorchPolicy::TorchPolicy

(

const char *

filename

)

Definition at line 38 of file TorchPolicy.cpp.

~TorchPolicy()

Rcs::TorchPolicy::~TorchPolicy ( )

virtual

Definition at line 49 of file TorchPolicy.cpp.

Member Function Documentation

computeAction()

void Rcs::TorchPolicy::computeAction ( MatNd *  action, const MatNd *  observation  )

virtual

Compute the action according to the policy.

Parameters

[out]	action	matrix to store the action in
[in]	observation	current observed state

Implements Rcs::ControlPolicy.

Definition at line 64 of file TorchPolicy.cpp.

reset()

void Rcs::TorchPolicy::reset ( )

virtual

Reset internal state if any. The default implementation does nothing.

Reimplemented from Rcs::ControlPolicy.

Definition at line 54 of file TorchPolicy.cpp.

Member Data Documentation

module

torch::jit::script::Module Rcs::TorchPolicy::module

private

Definition at line 56 of file TorchPolicy.h.

---

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

• src/cpp/core/control/TorchPolicy.h
• src/cpp/core/control/TorchPolicy.cpp