Add support for FixedDropout, issue #324

Dobiasd · Mar 11, 2022 · 0b40d76 · 0b40d76
1 parent 9566764
commit 0b40d76
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/include/fdeep/import_model.hpp b/include/fdeep/import_model.hpp
@@ -1170,6 +1170,7 @@ inline layer_ptr create_layer(const get_param_f& get_param,
             {"BatchNormalization", create_batch_normalization_layer},
             {"Dropout", create_identity_layer},
             {"AlphaDropout", create_identity_layer},
+            {"FixedDropout", create_identity_layer},
             {"GaussianDropout", create_identity_layer},
             {"GaussianNoise", create_identity_layer},
             {"SpatialDropout1D", create_identity_layer},