コード例 #1
0
ファイル: ncf_main.py プロジェクト: apple2333cream/models 
 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
ファイル: ncf_main.py プロジェクト: apple2333cream/models 
 def pred_input_fn():
   return dataset.input_fn(
       False, distribution_utils.per_device_batch_size(batch_size, num_gpus),
       ncf_dataset)
コード例 #3
0
ファイル: ncf_main.py プロジェクト: Alenza/AlenzaSyntaxnet 
 def pred_input_fn():
     return dataset.input_fn(False,
                             per_device_batch_size(batch_size, num_gpus),
                             ncf_dataset)
コード例 #4
0
ファイル: ncf_main.py プロジェクト: Alenza/AlenzaSyntaxnet 
 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
ファイル: ncf_main.py プロジェクト: youlingwangzi/TensorFlow 
 def pred_input_fn():
   return dataset.input_fn(
       False, per_device_batch_size(batch_size, num_gpus), ncf_dataset)
コード例 #6
0
ファイル: ncf_main.py プロジェクト: z331565360/CapProNet_tf 
 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)