Speed of Custom RNN is SUPER SLOW

you could try this.

#include <torch/csrc/jit/update_graph_executor_opt.h>
//...
setGraphExecutorOptimize(false);
3 Likes