Home
last modified time | relevance | path

Searched refs:steps_per_epoch (Results 1 – 25 of 40) sorted by relevance

12

/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_generator.py43 steps_per_epoch=None, argument
120 steps_per_epoch = kwargs['steps']
129 if steps_per_epoch is None:
131 steps_per_epoch = training_utils.infer_steps_for_dataset(
132 data, steps_per_epoch, epochs=epochs, steps_name=steps_name)
135 generator, steps_per_epoch = convert_to_generator_like(
137 steps_per_epoch=steps_per_epoch,
145 steps_per_epoch, validation_data, validation_steps, mode,
162 data, steps_per_epoch)
170 steps_per_epoch=steps_per_epoch,
[all …]
Dtraining_arrays.py58 steps_per_epoch=None, argument
121 steps_per_epoch = kwargs.pop('steps')
134 if steps_per_epoch is None:
136 steps_per_epoch = training_utils.infer_steps_for_dataset(
137 inputs, steps_per_epoch, epochs=epochs, steps_name=steps_name)
141 _print_train_info(inputs, val_inputs, steps_per_epoch, verbose)
152 use_steps = is_dataset or steps_per_epoch is not None
171 steps_per_epoch)
173 num_samples_or_steps = steps_per_epoch
201 steps_per_epoch=steps_per_epoch,
[all …]
Dtraining_distributed.py56 steps_per_epoch=None, argument
69 steps_per_epoch, batch_size = (
71 model._distribution_strategy, first_x_value, steps_per_epoch,
74 batch_size, steps_per_epoch, x)
115 steps_per_epoch=steps_per_epoch,
129 steps_per_epoch=steps_per_epoch,
273 steps_per_epoch=None, argument
312 steps_per_epoch = training_utils.infer_steps_for_dataset(
313 dataset, steps_per_epoch, epochs, steps_name='steps_per_epoch')
315 steps_per_epoch is None):
[all …]
Dtraining_generator_test.py80 steps_per_epoch=5,
87 steps_per_epoch=5,
93 steps_per_epoch=5,
101 steps_per_epoch=5,
186 steps_per_epoch=5,
192 steps_per_epoch=5,
223 steps_per_epoch=5,
230 steps_per_epoch=5,
264 steps_per_epoch=2,
290 steps_per_epoch=10,
[all …]
Dtraining_dataset_test.py68 model.fit(iterator, epochs=1, steps_per_epoch=2, verbose=1)
74 epochs=1, steps_per_epoch=2, verbose=0,
81 epochs=1, steps_per_epoch=2, verbose=0,
92 steps_per_epoch=2,
100 epochs=1, steps_per_epoch=2, verbose=0)
130 model.fit(iterator, epochs=1, steps_per_epoch=3, verbose=0)
160 model.fit(dataset, epochs=1, steps_per_epoch=2, verbose=0,
162 model.fit(dataset, epochs=1, steps_per_epoch=2, verbose=0,
181 model.fit(dataset, epochs=1, steps_per_epoch=2, verbose=1)
186 model.fit(dataset, epochs=1, steps_per_epoch=2, verbose=0,
[all …]
Dtraining.py544 steps_per_epoch=None, argument
721 steps_per_epoch=steps_per_epoch,
745 steps_per_epoch=steps_per_epoch,
750 batch_size, steps_per_epoch, x)
759 steps_per_epoch=steps_per_epoch,
782 steps_per_epoch=steps_per_epoch,
805 steps=steps_per_epoch,
844 steps_per_epoch=steps_per_epoch,
871 steps_per_epoch=steps_per_epoch,
1378 steps_per_epoch=None, argument
[all …]
Dtraining_eager_test.py197 model.fit(iterator, steps_per_epoch=2, epochs=1, verbose=0,
202 model.fit(iterator, steps_per_epoch=2, epochs=1, verbose=0,
206 model.fit(iterator, steps_per_epoch=2, epochs=1, verbose=0,
228 model.fit_generator(numpy_iterator(), steps_per_epoch=3, epochs=1)
281 history = model.fit(iterator, epochs=1, steps_per_epoch=10)
Dfeature_columns_integration_test.py92 model.fit(ds, steps_per_epoch=1)
93 model.fit(ds, steps_per_epoch=1)
137 dnn_model.fit(ds, steps_per_epoch=1)
138 dnn_model.fit(ds, steps_per_epoch=1)
Dsequential_test.py130 steps_per_epoch = 10
149 model.fit(iterator, epochs=1, steps_per_epoch=steps_per_epoch)
175 model.fit(inputs, targets, epochs=10, steps_per_epoch=30)
194 steps_per_epoch=2,
Dtraining_test.py739 model.fit(iterator, epochs=1, steps_per_epoch=2)
749 model.fit(iterator, epochs=1, steps_per_epoch=2)
851 model.fit(data, steps_per_epoch=16)
1709 model.fit(inputs, targets, epochs=1, steps_per_epoch=2, verbose=0)
1715 epochs=1, steps_per_epoch=2, verbose=0,
1724 model.fit(inputs, targets, epochs=1, steps_per_epoch=2, verbose=0)
1730 epochs=1, steps_per_epoch=2, verbose=0,
1768 steps_per_epoch=2,
1786 steps_per_epoch=2,
1798 steps_per_epoch=2,
[all …]
/external/tensorflow/tensorflow/contrib/distribute/python/
Dkeras_utils_test.py84 steps_per_epoch = 5
90 steps_per_epoch=steps_per_epoch,
100 num_batch_call_per_epoch = steps_per_epoch // steps_per_run
101 if steps_per_epoch % steps_per_run:
104 num_batch_call_per_epoch = steps_per_epoch
244 steps_per_epoch=2,
257 steps_per_epoch=2,
308 steps_per_epoch=2,
318 steps_per_epoch=2,
445 hist = model.fit(x=dataset, epochs=1, steps_per_epoch=2)
[all …]
Dkeras_backward_compat_test.py441 steps_per_epoch=2, verbose=1)
485 model.fit(dataset, epochs=1, steps_per_epoch=2, verbose=0,
487 model.fit(dataset, epochs=1, steps_per_epoch=2, verbose=0,
513 dataset, epochs=2, steps_per_epoch=2, verbose=1,
520 dataset, epochs=1, steps_per_epoch=2, verbose=1, shuffle=False)
560 model.fit(dataset_tuple, epochs=1, steps_per_epoch=2, verbose=1)
569 model.fit(dataset_dict, epochs=1, steps_per_epoch=2, verbose=1)
583 model.fit(dataset, epochs=1, steps_per_epoch=2, verbose=1)
597 model.fit(dataset, epochs=1, steps_per_epoch=2, verbose=1)
616 model.fit(dataset, epochs=1, steps_per_epoch=2, verbose=1)
[all …]
Dkeras_test.py689 steps_per_epoch=2, verbose=1)
796 model.fit(dataset, epochs=1, steps_per_epoch=2, verbose=0,
798 model.fit(dataset, epochs=1, steps_per_epoch=2, verbose=0,
823 dataset, epochs=2, steps_per_epoch=2, verbose=1,
830 dataset, epochs=1, steps_per_epoch=2, verbose=1, shuffle=False)
876 model.fit(dataset_tuple, epochs=1, steps_per_epoch=2, verbose=1)
885 model.fit(dataset_dict, epochs=1, steps_per_epoch=2, verbose=1)
988 model.fit(dataset, epochs=1, steps_per_epoch=2, verbose=1)
1002 model.fit(dataset, epochs=1, steps_per_epoch=2, verbose=1)
1023 model.fit(dataset, epochs=1, steps_per_epoch=2, verbose=1)
[all …]
Dkeras_multi_worker_test.py58 def _mnist_synthetic_dataset(batch_size, steps_per_epoch): argument
60 x_train = array_ops.ones([batch_size * steps_per_epoch, 28, 28, 1],
62 y_train = array_ops.ones([batch_size * steps_per_epoch, 1],
273 model.fit(x=train_ds, epochs=2, steps_per_epoch=steps)
372 steps_per_epoch=steps,
433 steps_per_epoch=steps,
/external/tensorflow/tensorflow/python/debug/examples/
Ddebug_keras.py33 steps_per_epoch = 2
39 (xs, ys)).repeat(num_examples).batch(int(num_examples / steps_per_epoch))
57 model.fit(dataset, epochs=FLAGS.epochs, steps_per_epoch=steps_per_epoch)
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
Dkeras_support.py1486 steps_per_epoch=None, argument
1509 if steps_per_epoch is None:
1556 initial_epoch, steps_per_epoch, validation_steps, **kwargs)
1560 steps_per_epoch, validation_steps, **kwargs)
1616 sample_weight, initial_epoch, steps_per_epoch, argument
1621 if batch_size is None and steps_per_epoch is None:
1640 steps=steps_per_epoch,
1660 steps_per_epoch=steps_per_epoch,
1676 steps_per_epoch, argument
1689 if (steps_per_epoch is None and verbose and inputs and
[all …]
/external/tensorflow/tensorflow/examples/tutorials/mnist/
Dfully_connected_feed.py104 steps_per_epoch = data_set.num_examples // FLAGS.batch_size
105 num_examples = steps_per_epoch * FLAGS.batch_size
106 for step in xrange(steps_per_epoch):
/external/tensorflow/tensorflow/contrib/factorization/examples/
Dmnist.py107 steps_per_epoch = data_set.num_examples // batch_size
108 num_examples = steps_per_epoch * batch_size
109 for _ in xrange(steps_per_epoch):
/external/tensorflow/tensorflow/python/keras/
Dmetrics_correctness_test.py190 self._custom_generator(), steps_per_epoch=2, epochs=2)
311 self._custom_generator(), steps_per_epoch=2, epochs=2)
Dcallbacks.py56 steps_per_epoch=None, argument
104 steps_per_epoch=steps_per_epoch,
118 steps_per_epoch=None, argument
151 'steps': steps_per_epoch,
/external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/
Dmain_estimator_tpu.py228 steps_per_epoch = revnet_config.tpu_iters_per_epoch
269 revnet_config.max_train_iter / steps_per_epoch,
/external/tensorflow/tensorflow/contrib/distribute/python/examples/
Dkeras_mnist.py129 model.fit(x=train_ds, epochs=20, steps_per_epoch=468)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.keras.models.-model.pbtxt178 …shuffle\', \'class_weight\', \'sample_weight\', \'initial_epoch\', \'steps_per_epoch\', \'validati…
182 …argspec: "args=[\'self\', \'generator\', \'steps_per_epoch\', \'epochs\', \'verbose\', \'callbacks…
Dtensorflow.keras.-model.pbtxt178 …shuffle\', \'class_weight\', \'sample_weight\', \'initial_epoch\', \'steps_per_epoch\', \'validati…
182 …argspec: "args=[\'self\', \'generator\', \'steps_per_epoch\', \'epochs\', \'verbose\', \'callbacks…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.keras.models.-model.pbtxt178 …shuffle\', \'class_weight\', \'sample_weight\', \'initial_epoch\', \'steps_per_epoch\', \'validati…
182 …argspec: "args=[\'self\', \'generator\', \'steps_per_epoch\', \'epochs\', \'verbose\', \'callbacks…

12