Add more information (metrics, images) to an experiment (AimCallback is used there)

[andife]

❓Question

Hi,

I'm using the AimCallback (aim.keras import) for my training within tensorflow.

Now I wanta directly integrate some result so the same run. For example some metrics and images?
How can I combine an existing callback with additional infos?

Thank you

[mihran113]

Hey @andife! The instance of AimCallback class has a property named experiment, which is the underlying aim.Run object. So to track additional metrics and images, you'll need to do something like this:

from aim.keras import AimCallback
from aim import Image as AimImage

aim_callback = AimCallback()
...
aim_callback.experiment.track(0, 'test_metric')
aim_callback.experiment.track(AimImage(some_input), 'test_image_sequence')

You can find additional details about how to track images in the documentation, if there'll be any questions.
https://aimstack.readthedocs.io/en/latest/quick_start/supported_types.html#image-tracking-with-aim