droneProject
Loading...
Searching...
No Matches
Train_TDP.py File Reference

Go to the source code of this file.

Namespaces

namespace  Train_TDP
 

Variables

 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