PyTorch Tensorboard 與文檔中描述的不一樣

Question

我正在使用 PyTorch 的ùtils.tensorboard.writer來記錄 RNN 的訓練。 對於àdd_hparams()函數，文檔說：

參數：

• hparam_dict (dict) –字典中的每個鍵值對都是超參數的名稱及其對應的值。 值的類型可以是 bool、string、float、int 或 None 之一。
• metric_dict (dict) –字典中的每個鍵值對都是度量的名稱及其對應的值。 請注意，此處使用的密鑰在 tensorboard 記錄中應該是唯一的。 否則，您通過 add_scalar 添加的值將顯示在 hparam 插件中。 在大多數情況下，這是不需要的。
• hparam_domain_discrete – (Optional[Dict[str, List[Any]]]) 包含超參數名稱和它們可以保存的所有離散值的字典
• run_name (str) – 運行的名稱，作為 logdir 的一部分包含在內。 如果未指定，將使用當前時間戳。

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

但是當我嘗試使用run_name參數時，出現錯誤

類型錯誤：add_hparams() 得到了一個意外的關鍵字參數“run_name”

所以我查找了我導入的 writer.py 文件，發現原因是add_hparams()函數本身：

def add_hparams(self, hparam_dict, metric_dict):

我檢查了 PyTorch 的安裝，但它是最新的。 這是某種夜間功能嗎，如果是這樣，我如何下載每晚版本的手電筒？

Answer 1

您鏈接到的文檔來自 PyTorch 當前主分支的文檔。 所以是的，這是一個夜間功能。 穩定版 (1.6) 的文檔沒有提到add_hparams 。

你可以得到命令在夜間下載PyTorch這里選擇Preview (Nightly) ，而不是Stable 。