void cScenarioArmTrainMACE::SetupActorScale() { auto trainer = std::static_pointer_cast<cMACETrainer>(mTrainer); if (!trainer->HasInitModel()) { auto ctrl = GetController(); int state_size = trainer->GetInputSize(); int action_size = trainer->GetOutputSize(); Eigen::VectorXd input_offset = Eigen::VectorXd::Zero(state_size); Eigen::VectorXd input_scale = Eigen::VectorXd::Ones(state_size); ctrl->BuildNNInputOffsetScale(input_offset, input_scale); trainer->SetInputOffsetScale(input_offset, input_scale); Eigen::VectorXd output_offset = Eigen::VectorXd::Zero(action_size); Eigen::VectorXd output_scale = Eigen::VectorXd::Ones(action_size); ctrl->BuildNNOutputOffsetScale(output_offset, output_scale); trainer->SetOutputOffsetScale(output_offset, output_scale); } }
void KTPStream::NotifyDependentTasks() { UINT32 uiInputCount = GetInputSize(); for (UINT32 i = 0; i < uiInputCount; i++) { KTPTask* pTask = GetInputAt(i); if (pTask) { pTask->m_bNeedReset = true; } } UINT32 uiOutputCount = GetOutputSize(); for (UINT32 i = 0; i < uiOutputCount; i++) { KTPTask* pTask = GetOutputAt(i); if (pTask) { pTask->m_bNeedReset = true; } } }
void KTPStream::RemoveFromDependentTasks() { UINT32 uiInputCount = GetInputSize(); for (UINT32 i = 0; i < uiInputCount; i++) { KTPTask* pkTask = GetInputAt(i); if (NULL != pkTask) { pkTask->RemoveInputImp(this); } } UINT32 uiOutputCount = GetOutputSize(); for (UINT32 i = 0; i < uiOutputCount; i++) { KTPTask* pkTask = GetOutputAt(i); if (NULL != pkTask) { pkTask->RemoveOutputImp(this); } } }
void StepNode::WriteMyOutputs(std::string& outCode) const { outCode += "\t" + VectorF::GetGLSLType(GetOutputSize(0)) + " " + GetOutputName(0) + " = step(" + GetInputs()[1].GetValue() + ", " + GetInputs()[0].GetValue() + ");\n"; }
void ReflectNode::WriteMyOutputs(std::string & outCode) const { outCode += "\t" + VectorF(GetOutputSize(0), 0).GetGLSLType() + " " + GetOutputName(0) + " = reflect(" + GetInputs()[0].GetValue() + ", " + GetInputs()[1].GetValue() + ");\n"; }
virtual int GetGradientsSize() const noexcept override OL_WARN_UNUSED_RESULT { return num_hyperparameters_*GetOutputSize(); }
int cTrainerInterface::GetActorOutputSize() const { return GetOutputSize(); }
int cTrainerInterface::GetCriticOutputSize() const { return GetOutputSize(); }