add hidden_layers_sizes to mimic EmbeddingIntentClassifier

RasaHQ · Mar 11, 2020 · baeee90 · baeee90
1 parent 68c2137
commit baeee90
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docs/migration-guide.rst b/docs/migration-guide.rst
@@ -90,6 +90,7 @@ General
     pipeline:
     # - ... other components
     - name: DIETClassifier
+      hidden_layers_sizes: {text: [256, 128]}
       intent_classification: True
       entity_recognition: False
       use_masked_language_model: False