---
title: tf.train.batch接口手工迁移-昇腾社区
description: TensorFlow不建议使用tf.train.batch接口通过队列形式处理输入数据。
keywords: tf.train.batch接口;队列形式
url: https://www.hiascend.com/document/caselibrary/detail/topic_0000001291311114
section: (其他)
---

# tf.train.batch接口手工迁移-昇腾社区

URL: https://www.hiascend.com/document/caselibrary/detail/topic_0000001291311114
描述: TensorFlow不建议使用tf.train.batch接口通过队列形式处理输入数据。
关键词: tf.train.batch接口;队列形式

昇腾文档 [了解详情](https://www.hiascend.com/document)

故障案例 [了解详情](https://www.hiascend.com/document/caselibrary)

tf.train.batch接口手工迁移

tf.train.batch接口手工迁移

2022/09/20

735

问题信息

| 问题来源 | 产品大类 | 产品子类 | 关键字 |
| --- | --- | --- | ---|
| 官方 | 模型训练 | TensorFlow | tf.train.batch接口、队列形式 |

#### 问题现象描述

TensorFlow不建议使用tf.train.batch接口通过队列形式处理输入数据。

#### 原因分析

TensorFlow不建议使用tf.train.batch接口通过队列形式处理输入数据，建议替换为tf.data.Dataset.batch(batch_size)；如果用户仍然希望在CANN上使用tf.train.batch接口，那么建议num_threads参数配置为1。

#### 解决措施

迁移示例如下所示：

原始脚本：

```
(texts, texts_tests, mels, mags, dones) = tf.train.batch([text, texts_test, mel, mag, done], shapes=[(hp.T_x,), (hp.T_x,), ((hp.T_y // hp.r), (hp.n_mels * hp.r)), (hp.T_y, (1 + (hp.n_fft // 2))), ((hp.T_y // hp.r),)], num_threads=2, batch_size=batch_size, capacity=(batch_size * 8), dynamic_pad=False)
```

迁移后脚本：

```
(texts, texts_tests, mels, mags, dones) = tf.train.batch([text, texts_test, mel, mag, done], shapes=[(hp.T_x,), (hp.T_x,), ((hp.T_y // hp.r), (hp.n_mels * hp.r)), (hp.T_y, (1 + (hp.n_fft // 2))), ((hp.T_y // hp.r),)], num_threads=1, batch_size=batch_size, capacity=(batch_size * 8), dynamic_pad=False)
```

本页内容

- 问题信息
- 问题现象描述
- 原因分析
- 解决措施