17.8.5 训练DDPG智能体执行自动驾驶任务
编写文件train_ddpg_agent.py,功能是训练一个DDPG智能体,以便在CARLA仿真环境中执行自动驾驶任务。它通过不断与环境互动,优化策略网络和值网络,以便智能体可以更好地执行驾驶任务。具体实现代码如下所示。
def train_agent(env, weather_list, agent, nb_training_episodes, save_folder, route_id, nb_updates=250, episode_skip=10):
if (agent.episode_nb+1)==1:
avg_reward, std_reward, success_rate = test_agent(env, weather_list, agent, route_id)
agent.tr_steps_vec.append(agent.tr_step+1)
agent.avg_reward_vec.append(avg_reward)