Skip to content

Commit e8ff8df

Browse files
dzelletensorflower-gardener
authored andcommitted
Flip drop_remainder default to False (to match tf.data.Dataset).
PiperOrigin-RevId: 481061799
1 parent a1ff905 commit e8ff8df

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tensorflow_gnn/runner/orchestration.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -313,8 +313,7 @@ def run(*,
313313
gtspec: tfgnn.GraphTensorSpec,
314314
global_batch_size: int,
315315
epochs: int = 1,
316-
# TODO(b/196880966): Flip this default to `False.`
317-
drop_remainder: bool = True,
316+
drop_remainder: bool = False,
318317
export_dirs: Optional[Sequence[str]] = None,
319318
model_exporters: Optional[Sequence[ModelExporter]] = None,
320319
feature_processors: Optional[Sequence[GraphTensorProcessorFn]] = None,

0 commit comments

Comments
 (0)