Commit Graph

  • eedc131d27 move if self.vqvae_model.check_discretized(): part inside def pretrain_vqvae jayLEE0301 2024-06-05 13:56:21 -0400
  • 1778dee9ab delete some short functions jayLEE0301 2024-06-05 13:47:06 -0400
  • 09b83d6584 Merge remote-tracking branch 'upstream/main' into train_tdmpc Alexander Soare 2024-06-05 18:41:47 +0100
  • 975da28461 change loss fn name, remove unnecessary resizing part jayLEE0301 2024-06-05 13:36:54 -0400
  • eaf0af0ac6 remove unused parts jayLEE0301 2024-06-05 13:09:37 -0400
  • be51483897 Fix artifacts with proper latency user/rcadene/2024_06_03_fix_delta_timestamps Remi Cadene 2024-06-05 16:56:08 +0000
  • 6b5c5b6dc4 Fix real-world configs by adding latency Remi Cadene 2024-06-03 14:47:06 +0000
  • 5d55b19cbd Fix tests Simon Alibert 2024-06-05 16:47:52 +0000
  • 641d349df4 Add save_image_transforms.py & artifacts Simon Alibert 2024-06-05 16:30:47 +0000
  • e444b0d529 Add first tests Simon Alibert 2024-06-05 16:29:54 +0000
  • 8237ed9aa4 Updated visualize script Marina Barannikov 2024-06-05 16:01:37 +0000
  • f69d552480 remove unnecessary resizing parts jayLEE0301 2024-06-05 11:57:41 -0400
  • 63d198cf0a change explanations jayLEE0301 2024-06-05 11:52:32 -0400
  • 36525b3d2e remove .cuda() parts jayLEE0301 2024-06-05 11:46:32 -0400
  • 75bdcaab81 remove unused parts in def discretize jayLEE0301 2024-06-05 11:36:10 -0400
  • 06d3eb40e1 remove redundant explanations jayLEE0301 2024-06-05 11:33:19 -0400
  • 78da71c31b add sequentially select options jayLEE0301 2024-06-05 11:29:08 -0400
  • dbc029b353 change confusing terms regarding action_tokens jayLEE0301 2024-06-05 11:23:24 -0400
  • 8c775c94fc simplified action stacking part jayLEE0301 2024-06-05 11:20:13 -0400
  • 82e32f1fcd Fix RandomSubsetApply weighted sampling Simon Alibert 2024-06-05 14:18:18 +0000
  • 644e77e413 Renamed scripts Marina Barannikov 2024-06-05 13:35:41 +0000
  • 1b1bbb1632 Minor formatting Marina Barannikov 2024-06-05 13:31:40 +0000
  • 0fb3dd745b Implented visualize_image_transforms script Marina Barannikov 2024-06-05 13:30:32 +0000
  • 4dbc1adb0d Updated show_transform to match config Marina Barannikov 2024-06-05 12:00:59 +0000
  • ceb95592af Remove prints Simon Alibert 2024-06-05 12:21:00 +0000
  • 6509c3f6d4 Implement RandomSubsetApply features Simon Alibert 2024-06-05 12:14:57 +0000
  • 8b134725d5
    Merge branch 'huggingface:main' into 2024_05_30_add_data_augmentation Marina Barannikov 2024-06-05 13:56:47 +0200
  • a544949ebe Added example of torchvision image augmentation on LeRobotDataset Marina Barannikov 2024-06-05 10:53:18 +0000
  • fdf56e7a62 Redesign config Simon Alibert 2024-06-05 09:49:31 +0000
  • 443b06b412 refactor show_image_transforms Simon Alibert 2024-06-05 09:34:39 +0000
  • 1eb4bfe2e4
    Fix videos_dir documentation (#247) Alexander Soare 2024-06-05 08:25:20 +0100
  • 55975a0334 fix videos_dir documentation Alexander Soare 2024-06-05 08:23:39 +0100
  • c4c5977f37 minor: nit jayLEE0301 2024-06-04 20:45:26 -0400
  • 57d3d27c78 add simple manual real world gym env example Thomas Wolf 2024-06-05 00:37:46 +0200
  • dd9906ebd9 update safetensors for all policies Akshay Kashyap 2024-06-04 15:27:05 -0700
  • c869930168 Merge branch 'main' into octo-model-poc Akshay Kashyap 2024-06-04 15:09:31 -0700
  • 6d72847bfe change n_action_pred_chunk -> action_chunk_size jayLEE0301 2024-06-04 17:59:49 -0400
  • bc9e6874fc replace forward part with concise one jayLEE0301 2024-06-04 17:41:34 -0400
  • 18b19b95f8 fix obs_projector to rgb_feature_projector, replace notimplementederror to assert jayLEE0301 2024-06-04 17:20:57 -0400
  • 9f109538d9 fix typo jayLEE0301 2024-06-04 17:14:26 -0400
  • 1f59e45f11 fix Phase -> Training Phase jayLEE0301 2024-06-04 17:12:46 -0400
  • 833d440ebf fix header -> head, bind pred -> bin prediction jayLEE0301 2024-06-04 17:10:39 -0400
  • 87842c0d19 add explanaitons for n_different_codes, n_different_combinations jayLEE0301 2024-06-04 17:05:04 -0400
  • 6012b4c859 change loss to loss_dict (vq-bet policy output name) jayLEE0301 2024-06-04 16:59:02 -0400
  • 0657c0c6c1 fix typo jayLEE0301 2024-06-04 16:53:49 -0400
  • c98d9dbd08 update traj generation + save tensors Akshay Kashyap 2024-06-04 13:14:52 -0700
  • 25914596c9
    Merge branch 'main' into local_logging_tensorboard dirkmcpherson 2024-06-04 15:49:25 -0400
  • 21f222fa1d
    Add out_dir option to eval (#244) Alexander Soare 2024-06-04 20:01:53 +0100
  • f88d0f0477 Merge remote-tracking branch 'origin/custom_eval_out_dir' into custom_eval_out_dir Alexander Soare 2024-06-04 18:39:01 +0100
  • 7cb6f2b759 revision Alexander Soare 2024-06-04 18:38:14 +0100
  • 5db9a96677 Merge remote-tracking branch 'upstream/main' into custom_eval_out_dir Alexander Soare 2024-06-04 18:25:12 +0100
  • 0935e49c8a add lerobotdataset in readme Thomas Wolf 2024-06-04 18:45:12 +0200
  • 33362dbd17
    Adding parameter dataloading_s to console logs and wandb for tracking… (#243) amandip7 2024-06-04 21:32:05 +0530
  • e4d03066cf
    Update lerobot/scripts/train.py amandip7 2024-06-04 19:46:51 +0530
  • 3c2dd1b881 fix test thomwolf_2024_06_04_nans Thomas Wolf 2024-06-04 15:43:10 +0200
  • e50b5a821c
    Merge branch 'main' into custom_eval_out_dir Alexander Soare 2024-06-04 14:10:33 +0100
  • 5e85a2c50b Add reachy2 dataset, policy, env Remi Cadene 2024-06-04 12:31:59 +0000
  • 22bd1f0669 Updated formatting Marina Barannikov 2024-06-04 12:06:36 +0000
  • 31e3c82386 Merge remote-tracking branch 'refs/remotes/origin/2024_05_30_add_data_augmentation' into 2024_05_30_add_data_augmentation Marina Barannikov 2024-06-04 12:00:46 +0000
  • 5eea2542d9 Added visualisations for image augmentation Marina Barannikov 2024-06-04 11:57:45 +0000
  • 42f9cc9c2a Updated transforms arguments Marina Barannikov 2024-06-04 11:14:54 +0000
  • 66629a956d Updated config to match transforms Marina Barannikov 2024-06-04 11:09:23 +0000
  • 7be2c35c0a
    Merge branch 'huggingface:main' into 2024_05_30_add_data_augmentation Simon Alibert 2024-06-04 12:28:06 +0200
  • b0d954c6e1
    Fix bug in normalize to avoid divide by zero (#239) Ruijie 2024-06-04 06:21:28 -0400
  • 63a5d0be39 fix nans Thomas Wolf 2024-06-04 11:40:46 +0200
  • e7737e79e3
    Merge branch 'main' into bug-normalize-zero Remi 2024-06-04 12:03:22 +0200
  • eac660bb9e fix nans thomwolf_2024_06_04_nan_fixes Thomas Wolf 2024-06-04 11:40:46 +0200
  • 1333560f6b Add dora_aloha_real_act_real and dora_aloha_real_act_real_no_state test artifacts Remi Cadene 2024-05-30 17:56:46 +0000
  • 92d1aecb40 Rename dora_aloha_real, WIP test_policies Remi Cadene 2024-05-30 17:54:59 +0000
  • 03d237fe0f small fix Remi Cadene 2024-05-30 13:38:19 +0000
  • 1c9f447ad0 small fix Remi Cadene 2024-05-30 13:36:34 +0000
  • 73f1a3932d Add dora-lerobot to pyproject Remi Cadene 2024-05-30 13:35:28 +0000
  • 97bda08e0f Rename Aloha2 to Aloha Remi Cadene 2024-05-30 13:35:02 +0000
  • 51dea3f67c fix Remi Cadene 2024-05-30 12:10:44 +0000
  • 5495d55cc7 Add aloha2_real, Add act_real, Fix vae=false, Add support for no state Remi Cadene 2024-05-30 12:06:57 +0000
  • 0c75f073a2 Add out_dir option to eval Alexander Soare 2024-06-04 08:20:56 +0100
  • 6d9a65ca39 remove check_discretized at VQBeTPolicy, and directly use the function in class VqVae jayLEE0301 2024-06-03 18:29:26 -0400
  • 651d9f46e5 remove .discretized, and add register_buffer jayLEE0301 2024-06-03 18:25:30 -0400
  • 26c8a48f82 Update gradio visualizer to used published gradio component Jeremy Leibs 2024-06-03 18:12:21 -0400
  • f0508d02b9 add self.reset() at the bottom of __init__ in class VQBeTPolicy jayLEE0301 2024-06-03 17:45:59 -0400
  • a33fbd4d44 SpatialSoftmax from modeling_diffusion.py jayLEE0301 2024-06-03 17:43:11 -0400
  • 8ed0181972 Updating console log entry from dloading_s to dataloading_s for consistency Git-bot9000 2024-06-03 23:21:50 +0530
  • 14291171cc Updated default.yaml Marina Barannikov 2024-06-03 17:21:16 +0000
  • cc4b3bd8e7 Updated default.yaml Marina Barannikov 2024-06-03 17:18:33 +0000
  • 602ea9844b Add RandomSubsetApply Simon Alibert 2024-06-03 17:15:37 +0000
  • 13e551e391
    Merge branch 'huggingface:main' into add_dataloading_s amandip7 2024-06-03 22:20:24 +0530
  • 330c535f57 Change 'best' backend to video_reader Simon Alibert 2024-06-03 16:09:02 +0000
  • a56626cf9c Add custom visualize_dataset.py Remi Cadene 2024-06-01 15:45:26 +0000
  • 44ba4ed566 Fix aloha (WIP: do not train in sim) Remi Cadene 2024-06-03 14:47:06 +0000
  • bd3111f28b
    Fix `visualize_dataset.py --help` (#241) Simon Alibert 2024-06-03 16:35:16 +0200
  • 9f8415fa83 Added clarification comments Marina Barannikov 2024-06-03 14:18:08 +0000
  • 6e46de6157 Merge remote-tracking branch 'origin/main' into user/aliberts/2024_05_28_compile_torchvision Simon Alibert 2024-06-03 13:41:28 +0000
  • 0ac3de5b88 Fix Simon Alibert 2024-06-03 15:24:34 +0200
  • cf15cba5fc
    Remove redundant slicing operation in Diffusion Policy (#240) Alexander Soare 2024-06-03 13:04:24 +0100
  • 0b847a558f
    Merge branch 'main' into update_push_to_hub Thomas Wolf 2024-06-03 11:52:42 +0200
  • f5ec659266 Remove redundant slicing operation in Diffusion Policy Alexander Soare 2024-06-03 10:46:26 +0100
  • 8b96a463f8
    Update lerobot/scripts/push_dataset_to_hub.py Thomas Wolf 2024-06-03 11:37:41 +0200
  • d67cab344c
    Update lerobot/scripts/push_dataset_to_hub.py Thomas Wolf 2024-06-03 11:37:35 +0200
  • b24ab712e3
    Update lerobot/scripts/push_dataset_to_hub.py Thomas Wolf 2024-06-03 11:37:29 +0200
  • 531e72616f
    Update lerobot/scripts/push_dataset_to_hub.py Thomas Wolf 2024-06-03 11:37:22 +0200