加载wandb 0.13.5时遇“Failed to sample metric”错误求助

排查环境依赖差异 #

• 对比本地与Colab的Python版本：wandb 0.13.5仅支持Python 3.7~3.10，若本地版本不在此范围，先调整Python版本。
• 导出双方依赖列表做对比：
  • Colab执行：pip freeze > colab_requirements.txt
  • 本地执行：pip freeze > local_requirements.txt
  • 重点核对numpy、torch、torchvision、yolox这些核心库的版本，确保本地版本与Colab一致，尤其是numpy版本差异容易引发wandb采样兼容问题。

修改wandb记录逻辑 #

• 手动转换metric类型：将YOLOX输出的tensor/numpy类型metric转为Python原生数值类型再传入wandb.log()，例如：

  wandb.log({"train_loss": float(train_loss.item()), "mAP": float(map_value)})
  

• 关闭自动系统采样：初始化wandb时禁用自动采集系统指标，避免不兼容的数据类型触发错误：

  import wandb
  wandb.init(
      project="your-yolox-project",
      settings=wandb.Settings(_disable_stats=True)
  )
  

清理wandb本地缓存 #

• 执行缓存清理命令：

  wandb cache clean
  

• 删除本地项目目录或用户根目录下的.wandb文件夹，重新初始化wandb后再启动训练。

调整wandb版本 #

• 尝试升级到0.14.x版本：

  pip install --upgrade wandb==0.14.0
  

• 或降级到0.13.4版本：

  pip install wandb==0.13.4
  

  同版本在不同环境下的兼容问题常源于依赖子库差异，调整版本可绕过特定版本的bug。

检查YOLOX的wandb集成代码 #

• 确认训练循环中wandb.log()传递的参数结构，避免嵌套字典、多维数组这类wandb无法解析的数据格式，将复杂metric扁平化处理后再记录。

内容的提问来源于stack exchange，提问作者Sebastien Grand