1 #pragma once 2 3 #include <torch/serialize/input-archive.h> 4 #include <torch/serialize/output-archive.h> 5