README.md
pyproject.toml
baselines/pi_policy.py
baselines/rum_policy.py
baselines/rum/__init__.py
baselines/rum/policy.py
baselines/rum/loss_fns/__init__.py
baselines/rum/loss_fns/abstract_loss_fn.py
baselines/rum/loss_fns/diffusion_policy_loss_fn.py
baselines/rum/loss_fns/rvq_loss_fn.py
baselines/rum/loss_fns/vqbet_loss_fn.py
baselines/rum/models/__init__.py
baselines/rum/models/custom.py
baselines/rum/models/bet/__init__.py
baselines/rum/models/bet/bet.py
baselines/rum/models/bet/gpt.py
baselines/rum/models/bet/tokenized_bet.py
baselines/rum/models/bet/utils.py
baselines/rum/models/bet/vqbet.py
baselines/rum/models/bet/vqvae/__init__.py
baselines/rum/models/bet/vqvae/residual_vq.py
baselines/rum/models/bet/vqvae/vector_quantize_pytorch.py
baselines/rum/models/bet/vqvae/vqvae.py
baselines/rum/models/bet/vqvae/vqvae_utils.py
baselines/rum/models/encoders/__init__.py
baselines/rum/models/encoders/abstract_base_encoder.py
baselines/rum/models/encoders/identity.py
baselines/rum/models/encoders/timm_encoders.py
baselines/rum/models/policies/diffusion_policy.py
baselines/rum/models/policies/open_loop.py
baselines/rum/models/policies/simple_open_loop.py
baselines/rum/molmo/server.py
baselines/rum/utils/__init__.py
baselines/rum/utils/action_transforms.py
baselines/rum/utils/decord_transforms.py
baselines/rum/utils/rpc.py
baselines/rum/utils/schedulers.py
baselines/rum/utils/trajectory_vis.py
baselines/rum/utils/zmq_utils.py
egogym/__init__.py
egogym/egogym.py
egogym/evaluate.py
egogym/utils.py
egogym.egg-info/PKG-INFO
egogym.egg-info/SOURCES.txt
egogym.egg-info/dependency_links.txt
egogym.egg-info/requires.txt
egogym.egg-info/top_level.txt
egogym/assets/constants.py
egogym/components/__init__.py
egogym/components/object.py
egogym/embodiments/__init__.py
egogym/embodiments/robot.py
egogym/embodiments/arms/__init__.py
egogym/embodiments/arms/arm.py
egogym/embodiments/arms/droid.py
egogym/embodiments/grippers/__init__.py
egogym/embodiments/grippers/floating_gripper.py
egogym/embodiments/grippers/rum.py
egogym/managers/__init__.py
egogym/managers/objects_managers.py
egogym/managers/textures_manager.py
egogym/misc/molmo_client.py
egogym/misc/molmo_server.py
egogym/policies/__init__.py
egogym/policies/base_policy.py
egogym/scripts/analayze.py
egogym/scripts/plot.py
egogym/scripts/plot_correlation.py
egogym/scripts/plot_correlation_hardcoded.py
egogym/scripts/plot_failure.py
egogym/scripts/plot_failure_hardcoded.py
egogym/scripts/plot_failure_vlm.py
egogym/scripts/plot_failure_vlm_hardcoded.py
egogym/scripts/plot_line.py
egogym/scripts/plot_line_hardcoded.py
egogym/scripts/plot_pi0_bars.py
egogym/tasks/close.py
egogym/tasks/open.py
egogym/tasks/pick.py
egogym/wrappers/__init__.py
egogym/wrappers/episode_monitor.py
egogym/wrappers/unprivileged_chatgpt.py
egogym/wrappers/unprivileged_gemini.py
egogym/wrappers/unprivileged_molmo.py
egogym/wrappers/unprivileged_moondream.py