PetaVision  Alpha
Public Member Functions | Protected Member Functions | Protected Attributes
PV::HebbianUpdater Class Reference
Inheritance diagram for PV::HebbianUpdater:
PV::BaseWeightUpdater PV::BaseObject PV::CheckpointerDataInterface PV::Observer PV::MomentumUpdater

BaseWeightUpdater Parameters

List of parameters needed from the BaseWeightUpdater class

virtual void ioParam_plasticityFlag (enum ParamsIOFlag ioFlag)
 plasticityFlag: Specifies if the weights will update
 

HebbianUpdater Parameters

List of parameters needed from the HebbianUpdater class

virtual void ioParam_triggerLayerName (enum ParamsIOFlag ioFlag)
 
virtual void ioParam_triggerOffset (enum ParamsIOFlag ioFlag)
 
virtual void ioParam_weightUpdatePeriod (enum ParamsIOFlag ioFlag)
 
virtual void ioParam_initialWeightUpdateTime (enum ParamsIOFlag ioFlag)
 
virtual void ioParam_immediateWeightUpdate (enum ParamsIOFlag ioFlag)
 
virtual void ioParam_dWMax (enum ParamsIOFlag ioFlag)
 
virtual void ioParam_dWMaxDecayInterval (enum ParamsIOFlag ioFlag)
 
virtual void ioParam_dWMaxDecayFactor (enum ParamsIOFlag ioFlag)
 
virtual void ioParam_normalizeDw (enum ParamsIOFlag ioFlag)
 
virtual void ioParam_useMask (enum ParamsIOFlag ioFlag)
 
virtual void ioParam_combine_dW_with_W_flag (enum ParamsIOFlag ioFlag)
 

Public Member Functions

 HebbianUpdater (char const *name, HyPerCol *hc)
 
void addClone (ConnectionData *connectionData)
 
bool getDataStructuresAllocatedFlag () const
 
float const * getDeltaWeightsDataHead (int arborId, int dataIndex) const
 
float const * getDeltaWeightsDataStart (int arborId) const
 
std::string const & getDescription () const
 
char const * getDescription_c () const
 
bool getInitialValuesSetFlag () const
 
bool getInitInfoCommunicatedFlag () const
 
MPIBlock const * getMPIBlock ()
 
char const * getName () const
 
std::string const & getObjectType () const
 
bool getPlasticityFlag () const
 
void ioParams (enum ParamsIOFlag ioFlag, bool printHeader, bool printFooter)
 
bool isUsingGPU () const
 
char const * lookupKeyword () const
 
void readParams ()
 
virtual Response::Status respond (std::shared_ptr< BaseMessage const > message) override
 
void writeParams ()
 

Protected Member Functions

virtual Response::Status allocateDataStructures () override
 
void blockingNormalize_dW ()
 
virtual Response::Status cleanup () override
 
int clear_dW (int arborId)
 
int clearNumActivations (int arborId)
 
virtual Response::Status communicateInitInfo (std::shared_ptr< CommunicateInitInfoMessage const > message) override
 
virtual void computeNewWeightUpdateTime (double time, double currentUpdateTime)
 
virtual Response::Status copyInitialStateToGPU ()
 
void decay_dWMax ()
 
int initialize (char const *name, HyPerCol *hc)
 
int initialize_dW (int arborId)
 
virtual Response::Status initializeState ()
 
int ioParamsFillGroup (enum ParamsIOFlag ioFlag) override
 
virtual bool needUpdate (double time, double dt)
 
virtual void normalize_dW ()
 
virtual int normalize_dW (int arbor_ID)
 
virtual Response::Status prepareCheckpointWrite () override
 
virtual Response::Status processCheckpointRead ()
 
virtual Response::Status readStateFromCheckpoint (Checkpointer *checkpointer) override
 
void reduce_dW ()
 
virtual int reduce_dW (int arborId)
 
void reduceAcrossBatch (int arborID)
 
virtual int reduceActivations (int arborID)
 
virtual int reduceKernels (int arborID)
 
virtual Response::Status registerData (Checkpointer *checkpointer) override
 
Response::Status respondAllocateData (std::shared_ptr< AllocateDataMessage const > message)
 
Response::Status respondCleanup (std::shared_ptr< CleanupMessage const > message)
 
Response::Status respondCommunicateInitInfo (std::shared_ptr< CommunicateInitInfoMessage const > message)
 
Response::Status respondCopyInitialStateToGPU (std::shared_ptr< CopyInitialStateToGPUMessage const > message)
 
Response::Status respondInitializeState (std::shared_ptr< InitializeStateMessage const > message)
 
Response::Status respondPrepareCheckpointWrite (std::shared_ptr< PrepareCheckpointWriteMessage const > message)
 
Response::Status respondProcessCheckpointRead (std::shared_ptr< ProcessCheckpointReadMessage const > message)
 
Response::Status respondReadStateFromCheckpoint (std::shared_ptr< ReadStateFromCheckpointMessage< Checkpointer > const > message)
 
Response::Status respondRegisterData (std::shared_ptr< RegisterDataMessage< Checkpointer > const > message)
 
Response::Status respondSetCudaDevice (std::shared_ptr< SetCudaDeviceMessage const > message)
 
virtual Response::Status setCudaDevice (std::shared_ptr< SetCudaDeviceMessage const > message)
 
void setDataStructuresAllocatedFlag ()
 
void setDescription ()
 
void setInitialValuesSetFlag ()
 
void setInitInfoCommunicatedFlag ()
 
int setName (char const *name)
 
virtual void setObjectType () override
 
int setParent (HyPerCol *hc)
 
int update_dW (int arborID)
 
void updateArbors ()
 
void updateInd_dW (int arborID, int batchID, float const *preLayerData, float const *postLayerData, int kExt)
 
void updateLocal_dW ()
 
virtual float updateRule_dW (float pre, float post)
 
virtual void updateState (double timestamp, double dt) override
 
virtual int updateWeights (int arborId)
 
void updateWeightsDelayed (double simTime, double dt)
 
void updateWeightsImmediate (double simTime, double dt)
 
void wait_dWReduceRequests ()
 

Protected Attributes

std::string description
 
ArborList * mArborList = nullptr
 
std::vector< ConnectionData * > mClones
 
bool mCombine_dWWithWFlag = false
 
ConnectionData * mConnectionData = nullptr
 
PVCuda::CudaDevice * mCudaDevice = nullptr
 
bool mDataStructuresAllocatedFlag = false
 
Weights * mDeltaWeights = nullptr
 
std::vector< MPI_Request > mDeltaWeightsReduceRequests
 
float mDWMax = std::numeric_limits<float>::quiet_NaN()
 
int mDWMaxDecayFactor = 0
 
float mDWMaxDecayInterval = 0.0f
 
int mDWMaxDecayTimer = 0
 
bool mImmediateWeightUpdate = true
 
bool mInitializeFromCheckpointFlag = false
 
bool mInitialValuesSetFlag = false
 
double mInitialWeightUpdateTime = 0.0
 
bool mInitInfoCommunicatedFlag = false
 
double mLastTimeUpdateCalled = 0.0
 
double mLastUpdateTime = 0.0
 
bool mNeedFinalize = true
 
bool mNormalizeDw = true
 
long ** mNumKernelActivations = nullptr
 
std::string mObjectType
 
bool mPlasticityFlag = true
 
bool mReductionPending = false
 
bool mTriggerFlag = false
 
HyPerLayer * mTriggerLayer = nullptr
 
char * mTriggerLayerName = nullptr
 
double mTriggerOffset = 0.0
 
bool mUsingGPUFlag = false
 
Weights * mWeights = nullptr
 
double mWeightUpdatePeriod = 0.0
 
double mWeightUpdateTime = 0.0
 
bool mWriteCompressedCheckpoints = false
 
char * name = nullptr
 
HyPerCol * parent = nullptr
 

Detailed Description

Definition at line 16 of file HebbianUpdater.hpp.

Member Function Documentation

void PV::HebbianUpdater::decay_dWMax ( )
protected

Decrements the counter for dWMaxDecayInterval, and if at the end of the interval, decays the dWMax value.

Definition at line 858 of file HebbianUpdater.cpp.

bool PV::BaseObject::getDataStructuresAllocatedFlag ( ) const
inlineinherited

Get-method for mDataStructuresAllocatedFlag, which is false on initialization and then becomes true once setDataStructuresAllocatedFlag() is called.

Definition at line 102 of file BaseObject.hpp.

bool PV::BaseObject::getInitialValuesSetFlag ( ) const
inlineinherited

Get-method for mInitialValuesSetFlag, which is false on initialization and then becomes true once setInitialValuesSetFlag() is called.

Definition at line 108 of file BaseObject.hpp.

bool PV::BaseObject::getInitInfoCommunicatedFlag ( ) const
inlineinherited

Get-method for mInitInfoCommunicatedFlag, which is false on initialization and then becomes true once setInitInfoCommunicatedFlag() is called.

Definition at line 95 of file BaseObject.hpp.

void PV::BaseObject::ioParams ( enum ParamsIOFlag  ioFlag,
bool  printHeader,
bool  printFooter 
)
inherited

Method for reading or writing the params from group in the parent HyPerCol's parameters. The group from params is selected using the name of the connection.

If ioFlag is set to write, the printHeader and printFooter flags control whether a header and footer for the parameter group is produces. These flags are set to true for layers, connections, and probes; and set to false for weight initializers and normalizers. If ioFlag is set to read, the printHeader and printFooter flags are ignored.

Note that ioParams is not virtual. To add parameters in a derived class, override ioParamFillGroup.

Definition at line 74 of file BaseObject.cpp.

int PV::HebbianUpdater::ioParamsFillGroup ( enum ParamsIOFlag  ioFlag)
overrideprotectedvirtual

The virtual method for reading parameters from the parent HyPerCol's parameters, and writing to the output params file.

Derived classes with additional parameters typically override ioParamsFillGroup to call the base class's ioParamsFillGroup method and then call ioParam_[parametername] for each of their parameters. The ioParam_[parametername] methods should call the parent HyPerCol's ioParamValue() and related methods, to ensure that all parameters that get read also get written to the outputParams-generated file.

Reimplemented from PV::BaseWeightUpdater.

Reimplemented in PV::MomentumUpdater.

Definition at line 27 of file HebbianUpdater.cpp.

bool PV::BaseObject::isUsingGPU ( ) const
inlineinherited

Returns true if the object requires the GPU; false otherwise. HyPerCol will not initialize the GPU unless one of the objects in its hierarchy returns true

Definition at line 116 of file BaseObject.hpp.

char const * PV::BaseObject::lookupKeyword ( ) const
inherited

Look up the keyword of the params group with the same name as the object.

Definition at line 42 of file BaseObject.cpp.

void PV::BaseObject::readParams ( )
inlineinherited

A method that reads the parameters for the group whose name matches the name of the object. It, along with writeParams(), is a wrapper around ioParams, so that readParams and writeParams automatically run through the same parameters in the same order.

Definition at line 62 of file BaseObject.hpp.

void PV::BaseObject::setDataStructuresAllocatedFlag ( )
inlineprotectedinherited

This method sets mDataStructuresAllocatedFlag to true.

Definition at line 175 of file BaseObject.hpp.

void PV::BaseObject::setInitialValuesSetFlag ( )
inlineprotectedinherited

This method sets the flag returned by getInitialValuesSetFlag to true.

Definition at line 180 of file BaseObject.hpp.

void PV::BaseObject::setInitInfoCommunicatedFlag ( )
inlineprotectedinherited

This method sets mInitInfoCommunicatedFlag to true.

Definition at line 170 of file BaseObject.hpp.

void PV::HebbianUpdater::updateLocal_dW ( )
protected

updateLocal_dW computes the contribution of the current process to dW, before MPI reduction and normalization. The routine calls initialize_dW for each arbor, and then updateWeights for each arbor.

Definition at line 422 of file HebbianUpdater.cpp.

void PV::BaseObject::writeParams ( )
inlineinherited

A method that writes the parameters for the group whose name matches the name of the object. It, along with readParams(), is a wrapper around ioParams, so that readParams and writeParams automatically run through the same parameters in the same order.

Definition at line 69 of file BaseObject.hpp.

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