Mava
Mava copied to clipboard
[FEATURE] Move tf.function in MAPPO
Feature
For a small performance boost we should consider moving the tf.functions from _minibatch_update to _step.
Testing
Validate that the trainer is running faster with this change. To do this we can use more than one executor so that the trainer does not wait for the executor data.