Use Tensor Cores for forward-inference in conv-LSTM

pull/749/head^2
AlexeyAB 6 years ago
parent 4b1ecfb419
commit 0ae5191580
  1. 2
      src/conv_lstm_layer.c

@ -799,7 +799,7 @@ void forward_conv_lstm_layer_gpu(layer l, network_state state)
s.train = state.train;
s.workspace = state.workspace;
s.net = state.net;
//if (!state.train) s.index = state.index; // don't use TC for training (especially without cuda_convert_f32_to_f16() )
if (!state.train) s.index = state.index; // don't use TC for training (especially without cuda_convert_f32_to_f16() )
int i;
layer vf = *(l.vf);
layer vi = *(l.vi);

Loading…
Cancel
Save