pytorch1.8复现TSN踩坑记录(如何在源代码中进行过更改)
最新推荐文章于 2024-05-15 09:46:01 发布
本文详细记录了在PyTorch 1.8中复现TSN模型时遇到的四个主要问题及其解决方案:1) 解决RuntimeError关于legacy autograd函数的警告;2) 将view替换为reshape以避免不兼容错误;3) 修正索引错误,避免对0-dim tensor的无效索引;4) 调整topk参数以解决维度范围错误。此外,还提到了在运行test_models.py时遇到的两个问题:1) 修复DataParallel的device_ids问题;2) 管理CUDA内存分配,避免内存不足。
摘要由CSDN通过智能技术生成