PyTorch Tensorboard 与文档中的描述不符

Posted 2023-03-27

【中文标题】PyTorch Tensorboard 与文档中的描述不符

【英文标题】：PyTorch Tensorboard not as described in documentation

【发布时间】：2020-12-21 17:06:28

【问题描述】：

我正在使用 PyTorch 的 ùtils.tensorboard.writer 记录 RNN 的训练。对于àdd_hparams() 函数，文档说：

参数：

hparam_dict (dict) - 字典中的每个键值对都是超参数的名称及其对应的值。值的类型可以是 bool、string、float、int 或 None 之一。 metric_dict (dict) – 字典中的每个键值对都是度量的名称及其对应的值。注意这里使用的键在张量板记录中应该是唯一的。否则，您通过 add_scalar 添加的值将显示在 hparam 插件中。在大多数情况下，这是不需要的。 hparam_domain_discrete – (Optional[Dict[str, List[Any]]]) 包含超参数名称和它们可以保存的所有离散值的字典 run_name (str) – 运行的名称，作为 logdir 的一部分。如果未指定，将使用当前时间戳。

来源：https://pytorch.org/docs/master/tensorboard.html

但是当我尝试使用run_name 参数时，我得到了错误

TypeError: add_hparams() 得到了一个意外的关键字参数“run_name”

于是我查看了我导入的 writer.py 文件，发现原因是 add_hparams() 函数本身：

def add_hparams(self, hparam_dict, metric_dict):

我检查了我的 PyTorch 安装，但它是最新的。这是某种夜间功能吗？如果是，我如何下载火炬的夜间版本？

【参考方案1】：

您链接到的文档是来自 PyTorch 当前主分支的文档。所以是的，这是一个夜间功能。稳定版（1.6）的文档没有提到add_hparams。

您可以通过选择 Preview (Nightly) 而不是 Stable 来获取下载 PyTorch 每晚 here 的命令。