Could wandb be used in the inference?

Could wandb be used in the inference stage? I’m specifically interested in collecting system metrics during inference. I’m working with DeepSpeed and torch profile and would to know if wandb could be useful in the inference as well.

Hi @eneas-jr,

Yes, wandb can definitely be used at inference time as well to record system metrics. wandb will record system metrics in between the calls to wandb.init() and wandb.finish().

Thanks,
Ramit

1 Like

Thanks a lot, @ramit_goolry

This topic was automatically closed 60 days after the last reply. New replies are no longer allowed.