Add an attention map image drawing function to the wandb log.

[sooftware]

🚀 Add an attention map image drawing function to the wandb log.

Example:

Motivation

To observe model training process.

Your contribution