Skip to content
Snippets Groups Projects
Commit 082fa660 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

improve intellisense

parent 452ba28d
No related branches found
No related tags found
No related merge requests found
......@@ -151,14 +151,14 @@ def fit(
validation_data = None
if strategy_fn is None:
model = model_fn()
model: tf.keras.Model = model_fn()
x = train_input_fn()
if eval_input_fn is not None:
validation_data = eval_input_fn()
else:
strategy = strategy_fn()
with strategy.scope():
model = model_fn()
model: tf.keras.Model = model_fn()
x = strategy.distribute_datasets_from_function(train_input_fn)
if eval_input_fn is not None:
validation_data = strategy.distribute_datasets_from_function(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment