Simon Alibert
|
8047a78301
|
rtol=50
|
2024-05-03 11:22:58 +02:00 |
|
Simon Alibert
|
a5a1d18362
|
Update artifacts after rebase from main
|
2024-05-03 11:08:26 +02:00 |
|
Simon Alibert
|
4bde95f93a
|
Disable test on tdmpc
|
2024-05-03 11:08:04 +02:00 |
|
Simon Alibert
|
75dd478b54
|
Revert "Revert "Override action queue defaults""
This reverts commit 5647d71c08.
|
2024-05-03 11:05:59 +02:00 |
|
Simon Alibert
|
5647d71c08
|
Revert "Override action queue defaults"
This reverts commit 7592c21eb2751c4ecc0fc1badb437f34b0f2155e.
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
61f38da7a3
|
Override action queue defaults
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
d08ff9e9a0
|
rename make_optimizer
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
e2618bd674
|
Add back full testing
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
6f11b0afaf
|
Cleanup
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
1acfd61b88
|
Dial back to atol=1e-7
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
97ded04b07
|
Try atol=1e-8
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
82b5ffb250
|
Try rtol=0.1
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
90cb75acfd
|
atol=1e-7
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
590b0eb48f
|
Add require_x86_64_kernel
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
55ff23c252
|
rtol=0.5
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
2d11199320
|
rtol=1, atol=1e-6 for all asserts
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
8f54825b1f
|
rtol=1, atol=1e-6
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
8e842ac00c
|
rtol=1e-4, atol=1e-7
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
e75448c8c4
|
Generate artifacts from x86 cpu
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
2ca594e801
|
atol = 1e-3
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
ff2cc0d7b1
|
Increase tolerance further again
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
53ae422eb8
|
Increase tolerance further
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
c3e57c2ae7
|
Increase tolerance
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
a66031f1f7
|
Remove default params torch.isclose
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
14648f81d0
|
Update cpu-generated artifact
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
77ee567851
|
Add require_cpu
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
62c1b9149a
|
Update artifacts
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
344c1653f2
|
Fix device
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
28d4122929
|
Create artifacts with DATA_DIR=tests/data
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
326933f208
|
Remove require_cuda
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
b0db762236
|
Update test artifacts
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
5877eb38ab
|
Enable backward compatibility testing on tdmpc
|
2024-05-03 10:04:49 +02:00 |
|
Simon Alibert
|
3a918b980f
|
WIP
|
2024-05-03 10:04:48 +02:00 |
|
Simon Alibert
|
276d210380
|
Add require_cuda
|
2024-05-03 10:04:48 +02:00 |
|
Simon Alibert
|
c1ecfdc663
|
Move to test_policies
|
2024-05-03 10:04:48 +02:00 |
|
Simon Alibert
|
30b32814a9
|
Add artifacts
|
2024-05-03 10:04:48 +02:00 |
|
Simon Alibert
|
7002813bbb
|
Cleanup, add output_dict
|
2024-05-03 10:04:48 +02:00 |
|
Simon Alibert
|
7ee03afabb
|
Test backward_compatibility
|
2024-05-03 10:04:48 +02:00 |
|
Simon Alibert
|
2726d1a6c9
|
Add get_policy_stats
|
2024-05-03 10:04:48 +02:00 |
|
Simon Alibert
|
6e15e9664b
|
WIP Add save_policy_to_safetensors
|
2024-05-03 10:04:48 +02:00 |
|
Simon Alibert
|
d84e37803f
|
WIP add test_backward_compatibility
|
2024-05-03 10:04:48 +02:00 |
|
Simon Alibert
|
14f80f294b
|
WIP Add test_backward
|
2024-05-03 10:04:48 +02:00 |
|
Simon Alibert
|
a541a7f3cf
|
Add make_optimizer
|
2024-05-03 10:04:48 +02:00 |
Remi
|
b2cda12f87
|
Add video decoding to LeRobotDataset (#92)
|
2024-05-03 00:50:19 +02:00 |
Alexander Soare
|
c1668924ab
|
Fix missing `policy.to(device)` in policy factory (#126)
|
2024-05-01 17:26:58 +01:00 |
|
Alexander Soare
|
d1855a202a
|
Refactor TD-MPC (#103)
Co-authored-by: Cadene <re.cadene@gmail.com>
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
|
2024-05-01 16:40:04 +01:00 |
|
Alexander Soare
|
a4891095e4
|
Use PytorchModelHubMixin to save models as safetensors (#125)
Co-authored-by: Remi <re.cadene@gmail.com>
|
2024-05-01 16:17:18 +01:00 |
|
Alexander Soare
|
01d5490d44
|
Follow `transformers` single file naming conventions (#124)
|
2024-05-01 13:09:42 +01:00 |
|
Alexander Soare
|
986583dc5c
|
Nest ACT model in ACT Policy (#122)
|
2024-04-30 17:17:40 +01:00 |
|
Alexander Soare
|
9d60dce6f3
|
Tidy up yaml configs (#121)
|
2024-04-30 16:08:59 +01:00 |