diff --git a/torch_training/render_agent_behavior.py b/torch_training/render_agent_behavior.py index bfe5ba030d265fa1f765946d418dbd9e526558b8..669f1c06c65346e23a902b47ffdac9ce7af73d43 100644 --- a/torch_training/render_agent_behavior.py +++ b/torch_training/render_agent_behavior.py @@ -38,7 +38,7 @@ min_dist = 5 observation_builder = TreeObsForRailEnv(max_depth=2) # Use a the malfunction generator to break agents from time to time -stochastic_data = MalfunctionParameters(malfunction_rate=10000, # Rate of malfunction occurence +stochastic_data = MalfunctionParameters(malfunction_rate=1./10000, # Rate of malfunction occurence min_duration=15, # Minimal duration of malfunction max_duration=50 # Max duration of malfunction )