fix GRU, add LSTM

Willena · Jun 7, 2017 · c04744d · c04744d
1 parent 59262f4
commit c04744d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/network.c b/src/network.c
@@ -125,8 +125,8 @@ char *get_layer_string(LAYER_TYPE a)
             return "rnn";
         case GRU:
             return "gru";
-		case LSTM:
-			return "lstm";
+        case LSTM:
+	    return "lstm";
         case CRNN:
             return "crnn";
         case MAXPOOL: