---
title: 预处理batch操作设置drop_remainder
description: "跟随训练脚本逻辑，找到数据预处理文件：official/vision/image_classification/resnet/imagenet_preprocessing.py。"
url: https://www.hiascend.com/document/detail/zh/TensorFlowCommercial/latest/migration/tfmigr2/tfmigr2_000094.html
sourcePath: /source/zh/TensorFlowCommercial/900/migration/tfmigr2/tfmigr2_000094.html
indexId: 95731ad08ee6de1505fd4cf029bc80113775651d0989a956a8bdd8a258977ed273
---
# 预处理batch操作设置drop_remainder

跟随训练脚本逻辑，找到数据预处理文件：official/vision/image_classification/resnet/imagenet_preprocessing.py。

在数据读取函数“input_fn”内部设置drop_remainder为True，该迁移点完成。

```
def input_fn(is_training,
             data_dir,
             batch_size,
             dtype=tf.float32,
             datasets_num_private_threads=None,
             parse_record_fn=parse_record,
             input_context=None,
             drop_remainder=False,
             tf_data_experimental_slack=False,
             training_dataset_cache=False,
             filenames=None):

"""
……
Returns:
  A dataset that can be used for iteration.
"""
drop_remainder=True

if filenames is None:
  filenames = get_filenames(is_training, data_dir)
dataset = tf.data.Dataset.from_tensor_slices(filenames)
```