手工迁移
简介
添加@tf.function修饰
设置NPU为默认设备
预处理batch动作设置drop_reminder
替换LossScaleOptimizer
训练循环下沉时设置NPU上的循环次数
分布式训练脚本适配(兼容单卡)
启动训练参数保持与单卡CPU形态一致
compat.v1脚本手工迁移