2021-12-21 08:39:05 +01:00
|
|
|
# Counters for sampling and training steps (env- and agent steps).
|
|
|
|
|
NUM_ENV_STEPS_SAMPLED = "num_env_steps_sampled"
|
|
|
|
|
NUM_AGENT_STEPS_SAMPLED = "num_agent_steps_sampled"
|
|
|
|
|
NUM_ENV_STEPS_TRAINED = "num_env_steps_trained"
|
|
|
|
|
NUM_AGENT_STEPS_TRAINED = "num_agent_steps_trained"
|
|
|
|
|
|
|
|
|
|
# Counters to track target network updates.
|
|
|
|
|
LAST_TARGET_UPDATE_TS = "last_target_update_ts"
|
|
|
|
|
NUM_TARGET_UPDATES = "num_target_updates"
|
|
|
|
|
|
2022-01-27 22:07:05 +01:00
|
|
|
# Performance timers (keys for Trainer._timers or metrics.timers).
|
2022-02-10 13:44:22 +01:00
|
|
|
TRAINING_ITERATION_TIMER = "training_iteration"
|
2021-12-21 08:39:05 +01:00
|
|
|
APPLY_GRADS_TIMER = "apply_grad"
|
|
|
|
|
COMPUTE_GRADS_TIMER = "compute_grads"
|
2022-01-27 22:07:05 +01:00
|
|
|
SYNCH_WORKER_WEIGHTS_TIMER = "synch_weights"
|
2021-12-21 08:39:05 +01:00
|
|
|
GRAD_WAIT_TIMER = "grad_wait"
|
|
|
|
|
SAMPLE_TIMER = "sample"
|
|
|
|
|
LEARN_ON_BATCH_TIMER = "learn"
|
|
|
|
|
LOAD_BATCH_TIMER = "load"
|
2022-01-27 22:07:05 +01:00
|
|
|
TARGET_NET_UPDATE_TIMER = "target_net_update"
|
|
|
|
|
|
|
|
|
|
# Deprecated: Use `SYNCH_WORKER_WEIGHTS_TIMER` instead.
|
|
|
|
|
WORKER_UPDATE_TIMER = "update"
|
