分类 机器学习项目测试 下的文章

  1. 首页
  2. 机器学习项目测试

pytorch 的set_detect_anomaly 导致训练速度异常慢的问题

永远不要在正式使用 AMP (GradScaler) 训练时开启 set_detect_anomaly(True)。 Anomaly Detection ...
  • admin
  • 2026 年 05 月 26 日
  • 暂无评论
    •

    pytorch项目需要注意的事项

    1、如果有的模型在字典里，需要提前用nn.ModuleDict(other_model_dict)进行包...

  • admin
  • 2025 年 03 月 06 日
  • 暂无评论

    • Tracking 如何增加热力图可视化

    1、参考MixFormer的vis-attn即可。使用hook钩子出attn矩阵（一般是dropout层）。2、稍微改改vis_attn_maps代码即可
  • admin
  • 2025 年 01 月 07 日
  • 暂无评论
    •

    【错误解决方案】ModuleNotFoundError: No module named ‘torchvision.models.utils

    原因一般是torchvision版本过高。将from torchvision.models.utils ...

  • admin
  • 2024 年 10 月 30 日
  • 暂无评论

    • 目标跟踪（OSTrack）模型变量提取——Visualizer

    Visualizer是一个辅助深度学习模型中Attention模块可视化的小工具，主要功能是帮助取出嵌套在模型深处的Attention Map（也可以提取...
  • admin
  • 2024 年 08 月 28 日
  • 暂无评论