Go to the source code of this file.
|
| | Train_TDP.model_path = os.path.join('Training', 'Saved Models', f'td3_drone_model{datetime.datetime.now().strftime("%Y_%m_%d_%H_%M_%S")}') |
| |
| | Train_TDP.log_path = os.path.join('Training', 'Logs', 'td3_drone_logs') |
| |
| | Train_TDP.env = DroneRobot() |
| |
| | Train_TDP.target_location |
| |
| | Train_TDP.debugMode |
| |
| | Train_TDP.n_actions = env.action_space.shape[-1] |
| |
| | Train_TDP.action_noise = OrnsteinUhlenbeckActionNoise(mean=np.zeros(n_actions), sigma=float(0.1) * np.ones(n_actions)) |
| |
| int | Train_TDP.episodes = 1000 |
| |
| int | Train_TDP.timesteps = env.steps_per_episode * episodes |
| |
| | Train_TDP.model = TD3("MlpPolicy", env, action_noise=action_noise, verbose=1, device='cuda', tensorboard_log=log_path) |
| |
| | Train_TDP.total_timesteps |
| |
| | Train_TDP.log_interval |
| |
| | Train_TDP.path |
| |