summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/utils/TorchUtils.h1
1 files changed, 1 insertions, 0 deletions
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>