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

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

..

act

Refactor TD-MPC (#103)

2024-05-01 16:40:04 +01:00

diffusion

Refactor TD-MPC (#103)

2024-05-01 16:40:04 +01:00

tdmpc

Refactor TD-MPC (#103)

2024-05-01 16:40:04 +01:00

factory.py

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

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

normalize.py

Use PytorchModelHubMixin to save models as safetensors (#125)

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

policy_protocol.py

Refactor TD-MPC (#103)

2024-05-01 16:40:04 +01:00

utils.py

ready for review

2024-04-12 11:36:52 +01:00