from torch2hpvm import compile_torch_module
from dnn import AlexNet

compile_torch_module(AlexNet(), (1, 3, 32, 32), "/tmp/alexnet", True)