Esempio n. 1
0
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);
	}
}
Esempio n. 2
0
	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;
			}
		}
	}
Esempio n. 3
0
	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);
			}
		}
	}
Esempio n. 4
0
void StepNode::WriteMyOutputs(std::string& outCode) const
{
    outCode += "\t" + VectorF::GetGLSLType(GetOutputSize(0)) + " " + GetOutputName(0) +
                   " = step(" + GetInputs()[1].GetValue() + ", " + GetInputs()[0].GetValue() + ");\n";
}
Esempio n. 5
0
void ReflectNode::WriteMyOutputs(std::string & outCode) const
{
    outCode += "\t" + VectorF(GetOutputSize(0), 0).GetGLSLType() + " " + GetOutputName(0) +
        " = reflect(" + GetInputs()[0].GetValue() + ", " + GetInputs()[1].GetValue() + ");\n";
}
Esempio n. 6
0
 virtual int GetGradientsSize() const noexcept override OL_WARN_UNUSED_RESULT {
   return num_hyperparameters_*GetOutputSize();
 }
Esempio n. 7
0
int cTrainerInterface::GetActorOutputSize() const
{
	return GetOutputSize();
}
Esempio n. 8
0
int cTrainerInterface::GetCriticOutputSize() const
{
	return GetOutputSize();
}