下载
EN
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

预处理batch动作设置drop_remainder

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

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

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
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)
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词