autocast

  • autocast:深度学习中的自动类型转换

    一、autocast介绍 autocast是PyTorch 1.6版本新增的一个功能,它是一个自动类型转换器,可以根据输入数据的类型自动选择合适的精度进行计算,从而使得计算速度更快…

    编程 2024-12-22