pytorch lightning 中使用tensorboard

from pytorch_lightning.loggers import TensorBoardLogger
logger = TensorBoardLogger('tb_logs', name='my_model')
trainer = Trainer(logger=logger)

 

你可能感兴趣的:(python,pytorch,pytorch)