protomotions.agents.utils package

Submodules

• protomotions.agents.utils.data module
  • swap_and_flatten01()
  • ExperienceBuffer
    • store_dict
    • __init__()
    • register_key()
    • update_data()
    • total_sum()
    • batch_update_data()
    • make_dict()
  • DictDataset
    • __init__()
    • shuffle()
    • num_batches()
• protomotions.agents.utils.metering module
  • Timer
    • time_total
    • num_ons
    • __init__()
    • on()
    • off()
    • report()
    • clear()
  • TimeReport
    • __init__()
    • add_timer()
    • start_timer()
    • end_timer()
    • report()
    • clear_timer()
    • pop_timer()
  • TensorAverageMeter
    • __init__()
    • add()
    • mean()
    • clear()
    • mean_and_clear()
  • TensorAverageMeterDict
    • __init__()
    • add()
    • mean()
    • clear()
    • mean_and_clear()
• protomotions.agents.utils.normalization module
  • RunningMeanStd
    • mean
    • var
    • count
    • __init__()
    • to()
    • maybe_clamp()
    • normalize()
  • combine_moments()
  • RewardRunningMeanStd
    • __init__()
    • record_reward()
    • normalize()
• protomotions.agents.utils.replay_buffer module
  • ReplayBuffer
    • _head
    • _is_full
    • __init__()
    • reset()
    • get_buffer_size()
    • store()
    • sample()
    • device
• protomotions.agents.utils.step_tracker module
  • StepTracker
    • steps
    • __init__()
    • advance()
    • done_indices()
    • reset_steps()
    • shift_counter()
    • device
• protomotions.agents.utils.training module
  • bounds_loss()
  • handle_model_grad_clipping()
  • aggregate_scalar_metrics()
  • get_activation_func()