init_with_imagenet parameters is False

gaborvecsei · gaborvecsei · commit d5eab8bc7b9e · 2019-08-04T09:08:06.000+02:00
diff --git a/fl.py b/fl.py
@@ -16,7 +16,7 @@
 
 
 def model_fn():
-    return fed_learn.create_model((32, 32, 3), 10)
+    return fed_learn.create_model((32, 32, 3), 10, init_with_imagenet=False)
 
 
 weight_summarizer = fed_learn.FedAvg()
