WarpDrive
Contents:
warp_drive package
example_envs package
WarpDrive
»
Index
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
K
|
L
|
M
|
N
|
O
|
P
|
R
|
S
|
T
|
U
|
V
|
W
A
A2C (class in warp_drive.training.algorithms.a2c)
ACTION_MASK (warp_drive.utils.constants.Constants attribute)
ACTIONS (warp_drive.utils.constants.Constants attribute)
add_data() (warp_drive.utils.data_feed.DataFeed method)
add_data_list() (warp_drive.utils.data_feed.DataFeed method)
add_meta_info() (data_manager.CUDADataManager method)
(warp_drive.managers.data_manager.CUDADataManager method)
add_shared_constants() (data_manager.CUDADataManager method)
(warp_drive.managers.data_manager.CUDADataManager method)
all_equal() (in module warp_drive.training.trainer)
(in module warp_drive.training.utils.data_loader)
apply_logit_mask() (in module warp_drive.training.models.fully_connected)
assign() (function_manager.CUDASampler static method)
(warp_drive.managers.function_manager.CUDASampler static method)
B
block (function_manager.CUDAFunctionManager property)
(warp_drive.managers.function_manager.CUDAFunctionManager property)
C
check_env_header() (in module warp_drive.utils.common)
compile (function_manager.CUDAFunctionManager property)
(warp_drive.managers.function_manager.CUDAFunctionManager property)
compile_and_load_cuda() (function_manager.CUDAFunctionManager method)
(warp_drive.managers.function_manager.CUDAFunctionManager method)
compute_distance() (example_envs.tag_continuous.tag_continuous.TagContinuous method)
compute_loss_and_metrics() (warp_drive.training.algorithms.a2c.A2C method)
(warp_drive.training.algorithms.ppo.PPO method)
compute_reward() (example_envs.tag_continuous.tag_continuous.TagContinuous method)
Constants (class in warp_drive.utils.constants)
create_and_push_data_placeholders() (in module warp_drive.training.utils.data_loader)
cuda_function_names (function_manager.CUDAFunctionManager property)
(warp_drive.managers.function_manager.CUDAFunctionManager property)
CUDADataManager (class in data_manager)
(class in warp_drive.managers.data_manager)
CUDAEnvironmentReset (class in function_manager)
(class in warp_drive.managers.function_manager)
CUDAFunctionFeed (class in function_manager)
(class in warp_drive.managers.function_manager)
CUDAFunctionManager (class in function_manager)
(class in warp_drive.managers.function_manager)
CUDALogController (class in function_manager)
(class in warp_drive.managers.function_manager)
CUDASampler (class in function_manager)
(class in warp_drive.managers.function_manager)
CUDATagGridWorld (class in example_envs.tag_gridworld.tag_gridworld)
CudaTensorHolder (class in data_manager)
(class in warp_drive.managers.data_manager)
D
data_manager
module
data_on_device_via_torch() (data_manager.CUDADataManager method)
(warp_drive.managers.data_manager.CUDADataManager method)
DataFeed (class in warp_drive.utils.data_feed)
device_data() (data_manager.CUDADataManager method)
(warp_drive.managers.data_manager.CUDADataManager method)
DONE_FLAGS (warp_drive.utils.constants.Constants attribute)
E
EnvironmentCPUvsGPU (class in warp_drive.env_cpu_gpu_consistency_checker)
EnvWrapper (class in warp_drive.env_wrapper)
example_envs
module
example_envs.tag_continuous
module
example_envs.tag_continuous.tag_continuous
module
example_envs.tag_gridworld
module
example_envs.tag_gridworld.tag_gridworld
module
F
fetch_episode_states() (warp_drive.training.trainer.Trainer method)
fetch_log() (function_manager.CUDALogController method)
(warp_drive.managers.function_manager.CUDALogController method)
forward() (warp_drive.training.models.fully_connected.FullyConnected method)
FullyConnected (class in warp_drive.training.models.fully_connected)
function_manager
module
G
generate_observation() (example_envs.tag_continuous.tag_continuous.TagContinuous method)
(example_envs.tag_gridworld.tag_gridworld.TagGridWorld method)
generate_random_actions() (in module warp_drive.env_cpu_gpu_consistency_checker)
get_data_dictionary() (example_envs.tag_continuous.tag_continuous.TagContinuous method)
(example_envs.tag_gridworld.tag_gridworld.CUDATagGridWorld method)
get_default_env_directory() (in module warp_drive.utils.common)
get_dtype() (data_manager.CUDADataManager method)
(warp_drive.managers.data_manager.CUDADataManager method)
get_flattened_obs() (warp_drive.training.models.fully_connected.FullyConnected method)
get_flattened_obs_size() (warp_drive.training.models.fully_connected.FullyConnected method)
get_function (function_manager.CUDAFunctionManager property)
(warp_drive.managers.function_manager.CUDAFunctionManager property)
get_obs() (in module warp_drive.training.utils.data_loader)
get_perf_stats() (warp_drive.training.trainer.PerfStats method)
get_project_root() (in module warp_drive.utils.common)
get_shape() (data_manager.CUDADataManager method)
(warp_drive.managers.data_manager.CUDADataManager method)
get_tensor_dictionary() (example_envs.tag_continuous.tag_continuous.TagContinuous method)
(example_envs.tag_gridworld.tag_gridworld.CUDATagGridWorld method)
graceful_close() (warp_drive.training.trainer.Trainer method)
grid (function_manager.CUDAFunctionManager property)
(warp_drive.managers.function_manager.CUDAFunctionManager property)
H
host_data (data_manager.CUDADataManager property)
(warp_drive.managers.data_manager.CUDADataManager property)
I
init_random() (function_manager.CUDASampler method)
(warp_drive.managers.function_manager.CUDASampler method)
initialize_default_functions() (function_manager.CUDAFunctionManager method)
(warp_drive.managers.function_manager.CUDAFunctionManager method)
initialize_functions() (function_manager.CUDAFunctionManager method)
(warp_drive.managers.function_manager.CUDAFunctionManager method)
initialize_shared_constants() (function_manager.CUDAFunctionManager method)
(warp_drive.managers.function_manager.CUDAFunctionManager method)
is_data_on_device() (data_manager.CUDADataManager method)
(warp_drive.managers.data_manager.CUDADataManager method)
is_data_on_device_via_torch() (data_manager.CUDADataManager method)
(warp_drive.managers.data_manager.CUDADataManager method)
K
k_nearest_neighbors() (example_envs.tag_continuous.tag_continuous.TagContinuous method)
L
load_cuda_from_binary_file() (function_manager.CUDAFunctionManager method)
(warp_drive.managers.function_manager.CUDAFunctionManager method)
load_cuda_from_source_code() (function_manager.CUDAFunctionManager method)
(warp_drive.managers.function_manager.CUDAFunctionManager method)
load_model_checkpoint() (warp_drive.training.trainer.Trainer method)
log_data_list (data_manager.CUDADataManager property)
(warp_drive.managers.data_manager.CUDADataManager property)
M
meta_info() (data_manager.CUDADataManager method)
(warp_drive.managers.data_manager.CUDADataManager method)
Metrics (class in warp_drive.training.trainer)
module
data_manager
example_envs
example_envs.tag_continuous
example_envs.tag_continuous.tag_continuous
example_envs.tag_gridworld
example_envs.tag_gridworld.tag_gridworld
function_manager
warp_drive
warp_drive.env_cpu_gpu_consistency_checker
warp_drive.env_wrapper
warp_drive.managers
warp_drive.managers.data_manager
warp_drive.managers.function_manager
warp_drive.training
warp_drive.training.algorithms
warp_drive.training.algorithms.a2c
warp_drive.training.algorithms.ppo
warp_drive.training.example_training_script
warp_drive.training.models
warp_drive.training.models.fully_connected
warp_drive.training.trainer
warp_drive.training.utils
warp_drive.training.utils.data_loader
warp_drive.utils
warp_drive.utils.common
warp_drive.utils.constants
warp_drive.utils.data_feed
warp_drive.utils.recursive_obs_dict_to_spaces_dict
N
name (example_envs.tag_continuous.tag_continuous.TagContinuous attribute)
(example_envs.tag_gridworld.tag_gridworld.TagGridWorld attribute)
(warp_drive.training.models.fully_connected.FullyConnected attribute)
O
obs_at_reset() (warp_drive.env_wrapper.EnvWrapper method)
OBSERVATIONS (warp_drive.utils.constants.Constants attribute)
P
PerfStats (class in warp_drive.training.trainer)
PPO (class in warp_drive.training.algorithms.ppo)
pretty_print() (warp_drive.training.trainer.Metrics method)
(warp_drive.training.trainer.PerfStats static method)
PROCESSED_OBSERVATIONS (warp_drive.utils.constants.Constants attribute)
pull_data_from_device() (data_manager.CUDADataManager method)
(warp_drive.managers.data_manager.CUDADataManager method)
push_data_to_device() (data_manager.CUDADataManager method)
(warp_drive.managers.data_manager.CUDADataManager method)
push_processed_obs_to_batch() (warp_drive.training.models.fully_connected.FullyConnected method)
R
recursive_merge_config_dicts() (in module warp_drive.training.trainer)
recursive_obs_dict_to_spaces_dict() (in module warp_drive.utils.recursive_obs_dict_to_spaces_dict)
register_actions() (function_manager.CUDASampler method)
(warp_drive.managers.function_manager.CUDASampler method)
reset() (example_envs.tag_continuous.tag_continuous.TagContinuous method)
(example_envs.tag_gridworld.tag_gridworld.TagGridWorld method)
(warp_drive.env_wrapper.EnvWrapper method)
reset_all_envs() (warp_drive.env_wrapper.EnvWrapper method)
reset_data_list (data_manager.CUDADataManager property)
(warp_drive.managers.data_manager.CUDADataManager property)
reset_device() (data_manager.CUDADataManager method)
(warp_drive.managers.data_manager.CUDADataManager method)
reset_log() (function_manager.CUDALogController method)
(warp_drive.managers.function_manager.CUDALogController method)
reset_only_done_envs() (warp_drive.env_wrapper.EnvWrapper method)
reset_when_done() (function_manager.CUDAEnvironmentReset method)
(warp_drive.managers.function_manager.CUDAEnvironmentReset method)
reset_when_done_deterministic() (function_manager.CUDAEnvironmentReset method)
(warp_drive.managers.function_manager.CUDAEnvironmentReset method)
reshape_and_flatten_obs() (warp_drive.training.models.fully_connected.FullyConnected method)
REWARDS (warp_drive.utils.constants.Constants attribute)
S
sample() (function_manager.CUDASampler method)
(warp_drive.managers.function_manager.CUDASampler method)
save_model_checkpoint() (warp_drive.training.trainer.Trainer method)
scalar_data_list (data_manager.CUDADataManager property)
(warp_drive.managers.data_manager.CUDADataManager property)
seed() (example_envs.tag_continuous.tag_continuous.TagContinuous method)
(example_envs.tag_gridworld.tag_gridworld.TagGridWorld method)
set_global_state() (example_envs.tag_continuous.tag_continuous.TagContinuous method)
(example_envs.tag_gridworld.tag_gridworld.TagGridWorld method)
setup_trainer_and_train() (in module warp_drive.training.example_training_script)
shared_constant() (data_manager.CUDADataManager method)
(warp_drive.managers.data_manager.CUDADataManager method)
step() (example_envs.tag_continuous.tag_continuous.TagContinuous method)
(example_envs.tag_gridworld.tag_gridworld.CUDATagGridWorld method)
(example_envs.tag_gridworld.tag_gridworld.TagGridWorld method)
(warp_drive.env_wrapper.EnvWrapper method)
step_all_envs() (warp_drive.env_wrapper.EnvWrapper method)
T
TagContinuous (class in example_envs.tag_continuous.tag_continuous)
TagGridWorld (class in example_envs.tag_gridworld.tag_gridworld)
test_env_reset_and_step() (warp_drive.env_cpu_gpu_consistency_checker.EnvironmentCPUvsGPU method)
train() (warp_drive.training.trainer.Trainer method)
Trainer (class in warp_drive.training.trainer)
training (warp_drive.training.models.fully_connected.FullyConnected attribute)
U
update_env_header() (in module warp_drive.utils.common)
update_env_runner() (in module warp_drive.utils.common)
update_log() (function_manager.CUDALogController method)
(warp_drive.managers.function_manager.CUDALogController method)
update_state() (example_envs.tag_continuous.tag_continuous.TagContinuous method)
(example_envs.tag_gridworld.tag_gridworld.TagGridWorld method)
V
verbose_print() (in module warp_drive.training.trainer)
W
warp_drive
module
warp_drive.env_cpu_gpu_consistency_checker
module
warp_drive.env_wrapper
module
warp_drive.managers
module
warp_drive.managers.data_manager
module
warp_drive.managers.function_manager
module
warp_drive.training
module
warp_drive.training.algorithms
module
warp_drive.training.algorithms.a2c
module
warp_drive.training.algorithms.ppo
module
warp_drive.training.example_training_script
module
warp_drive.training.models
module
warp_drive.training.models.fully_connected
module
warp_drive.training.trainer
module
warp_drive.training.utils
module
warp_drive.training.utils.data_loader
module
warp_drive.utils
module
warp_drive.utils.common
module
warp_drive.utils.constants
module
warp_drive.utils.data_feed
module
warp_drive.utils.recursive_obs_dict_to_spaces_dict
module