-
Notifications
You must be signed in to change notification settings - Fork 3.6k
Open
Labels
featureIs an improvement or enhancementIs an improvement or enhancementloggerRelated to the LoggersRelated to the Loggerstrainer: argument
Milestone
Description
🚀 Feature
When using multiple loggers, being able to give a dict of loggers to the Trainer() instead of a list.
Motivation
I find the current list access when calling a single logger not explicit.
Pitch
trainer = Trainer(
logger={
'tensorboard': TensorBoardLogger(...),
'csv': CSVLogger(...)},
...
)then in the loop:
self.logger['tensorboard'].experiment.add_image(...)instead of
self.logger[0].experiment.add_image(...)There could be an ambiguity when using an integer as key, so maybe enforce string-only keys.
cc @Borda @awaelchli @edward-io @ananthsub @rohitgr7 @kamil-kaczmarek @Raalsky @Blaizzy @justusschock @kaushikb11
F-Barto
Metadata
Metadata
Assignees
Labels
featureIs an improvement or enhancementIs an improvement or enhancementloggerRelated to the LoggersRelated to the Loggerstrainer: argument