昇腾社区首页
中文
注册

训练开发

训练开发(PyTorch)

  • PyTorch 2.1.0支持elastic分布式训练,支持自定义算子redispatch。
  • 优化内存池,通过虚拟内存机制降低设备内存占用率。
  • 支持模型异步保存。
  • 新增部分API支持及特性支持。

训练开发(TensorFlow)

  • TF Adapter 1.15 session配置和NPURunConfig构造函数中的enable_exception_dump参数新增“2:LiteExecptionDump”配置参数,用于dump异常算子的输入数据。
  • TF Adapter 2.6.5 npu.global_options函数中的enable_exception_dump参数新增“2:LiteExecptionDump”配置参数,用于dump异常算子的输入数据。