示例#1
0
 def train_input_fn():
   return dataset.input_fn(
       True,
       distribution_utils.per_device_batch_size(FLAGS.batch_size, num_gpus),
       ncf_dataset, FLAGS.epochs_between_evals)
示例#2
0
 def pred_input_fn():
   return dataset.input_fn(
       False, distribution_utils.per_device_batch_size(batch_size, num_gpus),
       ncf_dataset)
示例#3
0
 def pred_input_fn():
     return dataset.input_fn(False,
                             per_device_batch_size(batch_size, num_gpus),
                             ncf_dataset)
示例#4
0
 def train_input_fn():
     return dataset.input_fn(
         True, per_device_batch_size(FLAGS.batch_size, num_gpus),
         ncf_dataset, FLAGS.epochs_between_evals)
示例#5
0
 def pred_input_fn():
   return dataset.input_fn(
       False, per_device_batch_size(batch_size, num_gpus), ncf_dataset)
示例#6
0
 def pred_input_fn():
     return dataset.input_fn(
         False,
         distribution_utils.per_device_batch_size(batch_size, num_gpus),
         ncf_dataset)
示例#7
0
 def pred_input_fn():
     return dataset.input_fn(False,
                             per_device_batch_size(batch_size, num_gpus),
                             num_parallel_calls=num_parallel_calls)