Force rnn forward training

apache · Sep 26, 2019 · 025a227 · 025a227
1 parent 985a4ca
commit 025a227
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/operator/rnn-inl.h b/src/operator/rnn-inl.h
@@ -850,7 +850,7 @@ class RNNOp {
       }
       DType* work_cpu_space = static_cast<DType*>(temp_cpu_space_.data().dptr_);
 
-      if (ctx.is_train) {
+      if (1) {
         // allocate reserve space
 
         const size_t r_size = GetRNNReserveSpaceSize(param_.num_layers, direction,