参考链接: add_hparams
代码:
from torch.utils.tensorboard import SummaryWriter
import time
with SummaryWriter() as w:
for i in range(5):
w.add_hparams({'lr': 0.1*i, 'bsize': i},
{'超参数hparam/准确率accuracy': 10*i,
'超参数hparam/损失loss': 10*i})
print(time.time())
# tensorboard --logdir=runs
运行结果展示: