Python PyTorch SummaryWriter.add_custom_scalars用法及代碼示例

本文簡要介紹python語言中 torch.utils.tensorboard.writer.SummaryWriter.add_custom_scalars 的用法。

用法:

add_custom_scalars(layout)

參數：

layout(dict) -{分類名稱：圖表}， 在哪裏圖表也是字典 {chartName:ListOfProperties}。中的第一個元素ListOfProperties是圖表的類型(其中之一多行或者利潤) 並且第二個元素應該是一個列表，其中包含您在add_scalar 函數中使用的標簽，這些標簽將被收集到新圖表中。

通過收集‘scalars’中的圖表標簽來創建特殊圖表。請注意，每個SummaryWriter() 對象隻能調用一次此函數。因為它隻向 tensorboard 提供元數據，所以可以在訓練循環之前或之後調用該函數。

例子：

layout = {'Taiwan':{'twse':['Multiline',['twse/0050', 'twse/2330']]},
             'USA':{ 'dow':['Margin',   ['dow/aaa', 'dow/bbb', 'dow/ccc']],
                  'nasdaq':['Margin',   ['nasdaq/aaa', 'nasdaq/bbb', 'nasdaq/ccc']]}}

writer.add_custom_scalars(layout)