add build in check network

antoinedemathelin · antoinedemathelin · commit ff03fe4494f2 · 2021-07-02T18:48:19.000+02:00
diff --git a/adapt/utils.py b/adapt/utils.py
@@ -204,6 +204,7 @@ def check_network(network, copy=True,
             new_network = clone_model(network)
             if (hasattr(network, "input_shape") and
                 hasattr(new_network, "input_shape")):
+                new_network.build(input_shape=network.input_shape)
                 new_network.set_weights(network.get_weights())
         except Exception as e:
             if force_copy:
