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

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

pytorch项目需要注意的事项

1、如果有的模型在字典里，需要提前用nn.ModuleDict(other_model_dict)进行包装，否则将不会被注入模型参数，也无法自动net.c...
  • admin
  • 2025 年 03 月 06 日
  • 暂无评论
    •

    Tracking 如何增加热力图可视化

    1、参考MixFormer的vis-attn即可。使用hook钩子出attn矩阵（一般是dropout层...

  • admin
  • 2025 年 01 月 07 日
  • 2 条评论

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

    原因一般是torchvision版本过高。将from torchvision.models.utils import load_state_dict_fr...
  • admin
  • 2024 年 10 月 30 日
  • 4 条评论
    •

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

    Visualizer是一个辅助深度学习模型中Attention模块可视化的小工具，主要功能是帮助取出嵌套...

  • admin
  • 2024 年 08 月 28 日
  • 1 条评论

    • GPU性能实时检测推荐

    nvitop（重点推荐）一款交互式的 NVIDIA GPU 设备性能、资源和进程的实时监测工具。"nvitop" 在实时监控 GPU 设备资源和性能方面具...
  • admin
  • 2024 年 07 月 31 日
  • 1 条评论