speedup

speedup.py

from mlagents_envs.environment import UnityEnvironment
from mlagents_envs.envs.unity_gym_env import UnityToGymWrapper
from mlagents_envs.side_channel.engine_configuration_channel import EngineConfigurationChannel

engine_channel = EngineConfigurationChannel()

unity_env = UnityEnvironment(
    file_name=r"Simple/Assignment2",  # use the path to your built game here
    side_channels=[engine_channel],
    no_graphics=False,  # set to True to disable rendering, which can speed up training
)

engine_channel.set_configuration_parameters(time_scale=20.0)

env = UnityToGymWrapper(unity_env)

while True:
    obs = env.reset()
    done = False
    while not done:
        action = env.action_space.sample()
        obs, reward, done, info = env.step(action)
        print(obs, reward, done, info)

env.close()