world-modelz / dreamax

A scalable Dreamer implementation in JAX
MIT License
11 stars 2 forks source link

inference world model instants #2

Closed XMaster96 closed 2 years ago

XMaster96 commented 2 years ago

The World Model class always creates an optimizer's, optimizer's state's, and Value model. Witch are only required for the training case.

AC: Add support for a only inference world model instants with out optimizer's, optimizer's state's, and a Value model.