diff --git a/mobile_back_tflite/cpp/backend_tflite/embedding_utils.cc b/mobile_back_tflite/cpp/backend_tflite/embedding_utils.cc index a4760f1be..99e9109a3 100644 --- a/mobile_back_tflite/cpp/backend_tflite/embedding_utils.cc +++ b/mobile_back_tflite/cpp/backend_tflite/embedding_utils.cc @@ -25,6 +25,10 @@ bool TsEmbeddingParser::parse_pickle(const std::string& filename) { EMBEDDING_DIM * sizeof(float)); } + // Reverse both timesteps and embeddings before storing + std::reverse(timesteps.begin(), timesteps.end()); + std::reverse(embeddings.begin(), embeddings.end()); + // Store in maps timesteps_[num_timesteps] = std::move(timesteps); embeddings_[num_timesteps] = std::move(embeddings);