Skip to content

Commit 0027b14

Browse files
committed
Fixing some warnings
1 parent 78b406f commit 0027b14

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

lstm.c

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1770,7 +1770,7 @@ void lstm_train(lstm_model_t** model_layers, lstm_model_parameters_t *params,
17701770

17711771
if ( p > 0 ) {
17721772
--p;
1773-
while ( p >= 0 && p <= layers - 1 ) {
1773+
while ( p <= layers - 1 ) {
17741774
lstm_forward_propagate(model_layers[p],
17751775
cache_layers[p+1][e2]->probs,
17761776
cache_layers[p][e1],
@@ -1983,8 +1983,10 @@ void lstm_train(lstm_model_t** model_layers, lstm_model_parameters_t *params,
19831983
++i;
19841984
}
19851985

1986-
lstm_free_model(M_layers[p]);
1987-
lstm_free_model(R_layers[p]);
1986+
if ( params->optimizer == OPTIMIZE_ADAM ) {
1987+
lstm_free_model(M_layers[p]);
1988+
lstm_free_model(R_layers[p]);
1989+
}
19881990

19891991
lstm_free_model(gradient_layers_entry[p]);
19901992
lstm_free_model(gradient_layers[p]);

0 commit comments

Comments
 (0)