Skip to content

Commit 9ee80bf

Browse files
authored
Retrain Walker (#5911)
* reduce hidden nodes to 256 and retrain 30M steps * update demo file
1 parent 8a4c5d8 commit 9ee80bf

File tree

7 files changed

+13
-21
lines changed

7 files changed

+13
-21
lines changed
Binary file not shown.

Project/Assets/ML-Agents/Examples/Walker/Demos/ExpertWalker.demo.meta

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

Project/Assets/ML-Agents/Examples/Walker/Prefabs/Platforms/Platform.prefab

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -136,8 +136,13 @@ PrefabInstance:
136136
type: 3}
137137
propertyPath: m_Model
138138
value:
139-
objectReference: {fileID: 5022602860645237092, guid: 71fe1fe4f177841a1b4b85987277495c,
139+
objectReference: {fileID: 5022602860645237092, guid: b000e8fcd1edc454d9e6deb84abefbf9,
140140
type: 3}
141+
- target: {fileID: 3599283675615088294, guid: dd62a2a5a838943fba3d2cbb377a75b3,
142+
type: 3}
143+
propertyPath: m_DeterministicInference
144+
value: 0
145+
objectReference: {fileID: 0}
141146
- target: {fileID: 3599283675615088295, guid: dd62a2a5a838943fba3d2cbb377a75b3,
142147
type: 3}
143148
propertyPath: target

Project/Assets/ML-Agents/Examples/Walker/Scripts/WalkerAgent.cs

Lines changed: 4 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -84,8 +84,6 @@ public override void Initialize()
8484
m_JdController.SetupBodyPart(handR);
8585

8686
m_ResetParams = Academy.Instance.EnvironmentParameters;
87-
88-
SetResetParameters();
8987
}
9088

9189
/// <summary>
@@ -107,8 +105,6 @@ public override void OnEpisodeBegin()
107105
//Set our goal walking speed
108106
MTargetWalkingSpeed =
109107
randomizeWalkSpeedEachEpisode ? Random.Range(0.1f, m_maxWalkingSpeed) : MTargetWalkingSpeed;
110-
111-
SetResetParameters();
112108
}
113109

114110
/// <summary>
@@ -240,7 +236,10 @@ void FixedUpdate()
240236

241237
// b. Rotation alignment with target direction.
242238
//This reward will approach 1 if it faces the target direction perfectly and approach zero as it deviates
243-
var lookAtTargetReward = (Vector3.Dot(cubeForward, head.forward) + 1) * .5F;
239+
var headForward = head.forward;
240+
headForward.y = 0;
241+
// var lookAtTargetReward = (Vector3.Dot(cubeForward, head.forward) + 1) * .5F;
242+
var lookAtTargetReward = (Vector3.Dot(cubeForward, headForward) + 1) * .5F;
244243

245244
//Check for NaNs
246245
if (float.IsNaN(lookAtTargetReward))
@@ -292,16 +291,4 @@ public void TouchedTarget()
292291
{
293292
AddReward(1f);
294293
}
295-
296-
public void SetTorsoMass()
297-
{
298-
m_JdController.bodyPartsDict[chest].rb.mass = m_ResetParams.GetWithDefault("chest_mass", 8);
299-
m_JdController.bodyPartsDict[spine].rb.mass = m_ResetParams.GetWithDefault("spine_mass", 8);
300-
m_JdController.bodyPartsDict[hips].rb.mass = m_ResetParams.GetWithDefault("hip_mass", 8);
301-
}
302-
303-
public void SetResetParameters()
304-
{
305-
SetTorsoMass();
306-
}
307294
}
Binary file not shown.

Project/Assets/ML-Agents/Examples/Walker/TFModels/Walker.onnx.meta

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

config/ppo/Walker.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ behaviors:
1212
learning_rate_schedule: linear
1313
network_settings:
1414
normalize: true
15-
hidden_units: 512
15+
hidden_units: 256
1616
num_layers: 3
1717
vis_encode_type: simple
1818
reward_signals:

0 commit comments

Comments
 (0)