File tree Expand file tree Collapse file tree
tensorflow_gnn/docs/guide Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -286,11 +286,14 @@ def run(*,
286286 gtspec : tfgnn.GraphTensorSpec,
287287 global_batch_size : int ,
288288 epochs : int = 1 ,
289- drop_remainder : bool = True ,
289+ drop_remainder : bool = False ,
290290 export_dirs : Optional[Sequence[str ]] = None ,
291+ model_exporters : Optional[Sequence[ModelExporter]] = None ,
291292 feature_processors : Optional[Sequence[GraphTensorProcessorFn]] = None ,
292- valid_ds_provider : Optional[DatasetProvider] = None ):
293- pass
293+ valid_ds_provider : Optional[DatasetProvider] = None ,
294+ train_padding : Optional[GraphTensorPadding] = None ,
295+ valid_padding : Optional[GraphTensorPadding] = None ,
296+ tf_data_service_config : Optional[TFDataServiceConfig] = None ):
294297```
295298
296299The ` model_fn ` is expected to take a ` tfgnn.GraphTensorSpec ` and return a
You can’t perform that action at this time.
0 commit comments