Add image summaries to training #5
Description
Image summaries are saved only in validation, we should add another summary for training as well.
{{ message }}
Image summaries are saved only in validation, we should add another summary for training as well.