Remi Cadene
|
4c400b41a5
|
Improve log msg in train.py
|
2024-03-03 13:22:09 +00:00 |
|
Remi Cadene
|
0f2fa4d9ef
|
Add obs queue to pusht, Set n_obs_steps=2 for diffusion (Not fully tested)
|
2024-03-03 13:21:31 +00:00 |
|
Remi Cadene
|
cbbed590a9
|
Add mode to NormalizeTransform with mean_std or min_max (Not fully tested)
|
2024-03-03 13:19:02 +00:00 |
Simon Alibert
|
d3aae3111c
|
Merge pull request #5 from Cadene/user/aliberts/cpu_run
Add run on cpu-only compatibility
|
2024-03-03 13:06:39 +01:00 |
Simon Alibert
|
b859e89936
|
Fix for PR #5
|
2024-03-03 13:05:21 +01:00 |
|
Simon Alibert
|
b33ec5a630
|
Add run on cpu-only compatibility
|
2024-03-03 12:47:26 +01:00 |
|
Remi Cadene
|
48ded3dbc7
|
fix
|
2024-03-02 18:11:50 +00:00 |
|
Remi Cadene
|
661bda45ea
|
imagenet_norm: False
|
2024-03-02 17:18:58 +00:00 |
|
Remi Cadene
|
80785f8d0e
|
Small fix, Refactor diffusion, Diffusion runs (TODO: remove normalization in diffusion)
|
2024-03-02 17:04:39 +00:00 |
|
Remi Cadene
|
45b4ecb727
|
pre-commit run -a
|
2024-03-02 15:58:21 +00:00 |
|
Remi Cadene
|
1ae6205269
|
Add Normalize, non_blocking=True in tdmpc, tdmpc run (TODO: diffusion)
|
2024-03-02 15:53:29 +00:00 |
|
Remi Cadene
|
b5a2f460ea
|
fix bus error
|
2024-03-01 14:22:05 +00:00 |
|
Simon Alibert
|
fa7f473142
|
Merge pull request #4 from Cadene/user/aliberts/pusht_buffer_auto_download
Added pusht dataset auto-download
|
2024-03-01 15:20:30 +01:00 |
|
Simon Alibert
|
c1942d45d3
|
Fixes for PR #4
|
2024-03-01 14:59:05 +01:00 |
|
Simon Alibert
|
b862145e22
|
Added pusht dataset auto-download
|
2024-03-01 14:31:54 +01:00 |
|
Cadene
|
ca948c1e5b
|
fix zip strict=False
|
2024-03-01 00:45:23 +00:00 |
|
Cadene
|
ae050d2e94
|
Solve conflicts + pre-commit run -a
|
2024-02-29 23:31:32 +00:00 |
|
Cadene
|
0b9027f05e
|
Clean logging, Refactor
|
2024-02-29 23:21:27 +00:00 |
|
Simon Alibert
|
cb7b375526
|
Merge pull request #3 from Cadene/pre-commit
Style & Formatting
|
2024-02-29 21:48:19 +01:00 |
|
Simon Alibert
|
2c05b75f45
|
Fixes for PR #3
|
2024-02-29 21:46:41 +01:00 |
|
Simon Alibert
|
7e024fdce6
|
Ran pre-commit run --all-files
|
2024-02-29 13:37:48 +01:00 |
|
Simon Alibert
|
0e0e760e43
|
Added pre-commit
|
2024-02-29 12:27:30 +01:00 |
|
Remi
|
afb6b8679e
|
Merge pull request #2 from Cadene/install
Added poetry setup
|
2024-02-28 19:47:03 +01:00 |
|
Cadene
|
5798979e9c
|
comment
|
2024-02-28 18:31:54 +00:00 |
|
Simon Alibert
|
d5bfeadcf1
|
Added poetry setup
|
2024-02-28 19:12:56 +01:00 |
|
Cadene
|
ac90b9c3ee
|
Fix diffusion (rm transpose), Add prefetch
|
2024-02-28 17:45:01 +00:00 |
|
Cadene
|
cf5063e50e
|
Add diffusion policy (train and eval works, TODO: reproduce results)
|
2024-02-28 15:21:42 +00:00 |
|
Simon Alibert
|
f1708c8a37
|
install fix
|
2024-02-28 12:35:49 +01:00 |
|
Simon Alibert
|
98f8869743
|
WIP
|
2024-02-28 10:59:06 +01:00 |
|
Cadene
|
e543c9a42c
|
small fix %
|
2024-02-27 11:54:31 +00:00 |
|
Cadene
|
7df542445c
|
Small fix and improve logging message
|
2024-02-27 11:44:26 +00:00 |
|
Cadene
|
21670dce90
|
Refactor train, eval_policy, logger, Add diffusion.yaml (WIP)
|
2024-02-26 01:10:09 +00:00 |
|
Cadene
|
5a219fed6e
|
Refactor policy config
|
2024-02-25 18:26:44 +00:00 |
|
Cadene
|
b16c334825
|
Refactor configs to have env in seperate yaml + Fix training
|
2024-02-25 17:42:47 +00:00 |
|
Cadene
|
eec134d72b
|
small change
|
2024-02-25 14:20:14 +00:00 |
|
Cadene
|
ed80db2846
|
Sanitize cfg.env
|
2024-02-25 12:02:29 +00:00 |
|
Cadene
|
9b469c4232
|
Fix pusht.yaml
|
2024-02-25 11:49:39 +00:00 |
|
Cadene
|
0eb9b5d1a5
|
Sanitize cfg.wandb
|
2024-02-25 11:15:09 +00:00 |
|
Cadene
|
5812298290
|
fix style readme
|
2024-02-25 11:09:47 +00:00 |
|
Cadene
|
e765e26b0b
|
Sanitize cfg.policy, Fix skip_frame pusht.yaml
|
2024-02-25 11:09:02 +00:00 |
|
Cadene
|
fc4b98544b
|
Add tests to Readme
|
2024-02-25 10:52:31 +00:00 |
|
Cadene
|
6f5c731936
|
Rename test -> tests
|
2024-02-25 10:51:07 +00:00 |
|
Cadene
|
598bb496b0
|
Add policies/factory, Add test, Add _self_ in config
|
2024-02-25 10:50:23 +00:00 |
|
Cadene
|
64b5920e94
|
format
|
2024-02-24 18:19:18 +00:00 |
|
Cadene
|
aed02dc7c6
|
Add multithreading for video generation, Speed policy sampling
|
2024-02-24 18:18:39 +00:00 |
|
Cadene
|
591985c67d
|
Fix done in pusht, Fix --time in sbatch
|
2024-02-22 17:58:26 +00:00 |
|
Cadene
|
664cfb2023
|
Add sbatch.sh
|
2024-02-22 13:04:32 +00:00 |
|
Cadene
|
63d18475cc
|
fix simxarm factory
|
2024-02-22 13:04:24 +00:00 |
|
Cadene
|
96c53ad06f
|
remove comments
|
2024-02-22 12:15:14 +00:00 |
|
Cadene
|
e3643d6146
|
Wandb works, One output dir
|
2024-02-22 12:14:12 +00:00 |