Tensorboard SummaryWriter across multiple files


I’m trying to use Tensorboard in a multi-file project. I have a main.py file that runs the training loop, but I would like to log some values from other packages/modules as well. What is the best practice here? Do I create a SummaryWriter in each module? If so, wouldn’t this cause problems with from foo import * style imports?

Since I already use a customer logger from the logging library, I was considering wrapping SummaryWriter within this.