Fix: No Context TFF error for Fed Training

parent 86a4a0be
......@@ -20,7 +20,7 @@ def model_fn():
def federated_computation_new(train_dataset,test_dataset):
import tensorflow as tf
import tensorflow_federated as tff
#TFF must be imported here, to guarantee that the context is initialised
if(globals.INPUT_SPEC == None):
#should never reach this place because INPUT_SPEC is instantiated inside get_preprocessed_train_test_data.
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment