使用.detach()代替requires=False避免计算图错误

解决对一个对象多次BP更新时造成的RuntimeError: you can only change requires_grad flags of leaf variables.问题_解决对一个对象多次bp更新时造成的问题-CSDN博客

代码中存在一个特征需要参与辅助损失的计算,由于需要反复进行反向传播更新,计算图容易出错,所以用.requires_grad_(False)替代.detach()对比如.(torch.cuda.DoubleTensor)的张量进行处理

你可能感兴趣的:(深度学习,人工智能)