Mava icon indicating copy to clipboard operation
Mava copied to clipboard

[FEATURE] Move tf.function in MAPPO

Open DriesSmit opened this issue 2 years ago • 0 comments

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.

DriesSmit avatar Apr 15 '22 04:04 DriesSmit