update

examples/vm_sound_classification/step_3_train_model.py

@@ -161,7 +161,10 @@ def main():
         # num_labels=vocabulary.get_vocab_size(namespace="labels")
     )
     if not config.cls_head_param["num_labels"] == vocabulary.get_vocab_size(namespace="labels"):
-        raise AssertionError
+        raise AssertionError("expected num labels: {} instead of {}.".format(
+            vocabulary.get_vocab_size(namespace="labels"),
+            config.cls_head_param["num_labels"],
+        ))
     model = WaveClassifierPretrainedModel(
         config=config,
     )