Yes, you can just call wandb.log anytime manually like so
wandb.log({"my_metric":1.2}, step=trainer.global_step)
Passing the step is not mandatory but preferred when logging at different places.
1 Like