File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed
tensorflow_gnn/docs/guide Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -286,11 +286,14 @@ def run(*,
286
286
gtspec : tfgnn.GraphTensorSpec,
287
287
global_batch_size : int ,
288
288
epochs : int = 1 ,
289
- drop_remainder : bool = True ,
289
+ drop_remainder : bool = False ,
290
290
export_dirs : Optional[Sequence[str ]] = None ,
291
+ model_exporters : Optional[Sequence[ModelExporter]] = None ,
291
292
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 ):
294
297
```
295
298
296
299
The ` model_fn ` is expected to take a ` tfgnn.GraphTensorSpec ` and return a
You can’t perform that action at this time.
0 commit comments