Alexander Soare c1668924ab Fix missing policy.to(device) in policy factory (#126)

2024-05-01 17:26:58 +01:00

..

datasets

Tidy up yaml configs (#121)

2024-04-30 16:08:59 +01:00

envs

Move normalization to policy for act and diffusion (#90)

2024-04-25 11:47:38 +02:00

policies

Fix missing policy.to(device) in policy factory (#126)

2024-05-01 17:26:58 +01:00

utils

Remove warnings (#111)

2024-04-29 00:31:33 +02:00

logger.py

Use PytorchModelHubMixin to save models as safetensors (#125)

2024-05-01 16:17:18 +01:00