search

entity-neural-network / incubator

Collection of in-progress libraries for entity neural networks.
Apache License 2.0
29 stars 10 forks source link

video recording for evaluation environments using tensorboard #154

Closed Bam4d closed 2 years ago

Bam4d commented 2 years ago

Added an option for generating videos during evaluation runs: --eval-capture-videos

also allowed customization of number processes used for evaluation, so we are a bit more flexible there. --eval-processes

uses writer.add_video from tensorboard.

Videos end up in wandb looking like this and are linked to specific eval runs by global_step variable: image

Bam4d commented 2 years ago

closes #152