experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Raise error when calling .fit() w/ batch_size and a tf dataset

Browse Source
This commit is contained in:
Jonah Kohn 2020-07-22 16:16:29 -07:00 committed by GitHub
parent 513fff7cf6
commit 8f278b5c18
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/python/keras/engine/data_adapter.py
View File

@ -690,6 +690,7 @@ class DatasetAdapter(DataAdapter):
y=None,
sample_weights=None,
steps=None,
batch_size=None,
**kwargs):
super(DatasetAdapter, self).__init__(x, y, **kwargs)
# Note that the dataset instance is immutable, its fine to reuse the user
@ -699,7 +700,7 @@ class DatasetAdapter(DataAdapter):
# The user-provided steps.
self._user_steps = steps
self._validate_args(y, sample_weights, steps)
self._validate_args(y, sample_weights, steps, batch_size)
def get_dataset(self):
return self._dataset
@ -728,7 +729,7 @@ class DatasetAdapter(DataAdapter):
return (self._user_steps is None or
cardinality.cardinality(self._dataset).numpy() == self._user_steps)
def _validate_args(self, y, sample_weights, steps):
def _validate_args(self, y, sample_weights, steps, batch_size):
"""Validates `__init__` arguments."""
# Arguments that shouldn't be passed.
if not is_none_or_empty(y):
@ -737,6 +738,10 @@ class DatasetAdapter(DataAdapter):
if not is_none_or_empty(sample_weights):
raise ValueError("`sample_weight` argument is not supported when using "
"dataset as input.")
if batch_size is not None:
raise ValueError("`batch_size` argument must not be specified when "
"using dataset as input.")
if steps is None:
if _is_distributed_dataset(self._dataset):