diff --git a/demo/demo2/config.yaml b/demo/demo2/config.yaml
index 2bd539051b4dbcc7b65763599f69bf81402f6047..f62c257f21a1dd55fe0de85546b8518702eac9de 100644
--- a/demo/demo2/config.yaml
+++ b/demo/demo2/config.yaml
@@ -12,7 +12,7 @@ do_lower_case: True
 max_seq_len: 512
 
 batch_size: 4
-num_epochs: 0.1
+num_epochs: 2
 optimizer: "adam"
 learning_rate: 3e-5
 warmup_proportion: 0.1