mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

fix GRU, add LSTM

Browse Source
This commit is contained in:
Yao Lu 2017-06-06 17:22:23 -07:00
parent 59262f4c77
commit c04744d15e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/network.c
View File

@ -125,8 +125,8 @@ char *get_layer_string(LAYER_TYPE a)
return "rnn"; return "rnn";
case GRU: case GRU:
return "gru"; return "gru";
case LSTM: case LSTM:
return "lstm"; return "lstm";
case CRNN: case CRNN:
return "crnn"; return "crnn";
case MAXPOOL: case MAXPOOL: