模型训练----apex库报错IndexError: tuple index out of range

本文主要是介绍模型训练----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的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---