🚀 Feature

Expose the metric tracking context in the pytorch_ignite logger, allowing users to set the context arbitrarily rather than just the subset.

Motivation

Currently, the ignite logger only sets the subset in the context. It could be useful to be able to add further items to the context, such as the fold number in a cross validated training run.

Pitch

aim.pytorch_ignite.AimLogger should take context as an argument. This should then be used as the starting point in the log_metrics method instead of an empty dict.