diff --git a/src/utils/TorchUtils.h b/src/utils/TorchUtils.h
index da30acb..8b7a3f1 100644
--- a/src/utils/TorchUtils.h
+++ b/src/utils/TorchUtils.h
@@ -1,6 +1,7 @@
 #pragma once
 
 #include <string>
+#include <vector>
 #include <fstream>
 #include <sstream>
 #include <iomanip>