Fixing mising import

hpvm/test/epoch_dnn/main.py

@@ -2,6 +2,7 @@ import site
 from pathlib import Path
 from os import makedirs
 import numpy as np
+import sys
 
 import torch
 from torch2hpvm import BinDataset, ModelExporter