本文主要是介绍模型训练----apex库报错IndexError: tuple index out of range，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！
问题描述
在训练模型的过程中遇到了apex库的报错IndexError: tuple index out of range导致无法训练。在github查询后找到了解决方法
问题解决
需要修改/apex-master/apex/amp/utils.py这个文件的代码
从93行开始修改
if x in cache:cached_x = cache[x]next_functions_available = Falseif x.requires_grad and cached_x.requires_grad:if len(cached_x.grad_fn.next_functions) > 1:next_functions_available = True# Make sure x is actually cached_x's autograd parent.if next_functions_available and cached_x.grad_fn.next_functions[1][0].variable is not x:
在修改下边120行左右
elif x.requires_grad and cached_x.requires_grad and not next_functions_available:del cache[x]
最后重新运行一遍apex的安装程序，再次运行训练就不会报错了
 | 点击访问博客查看更多内容 |
|---|
这篇关于模型训练----apex库报错IndexError: tuple index out of range的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
