Export simple test model

export_test_model.py

def export_test():
    image = torch.ones((1, 3, 256, 256), requires_grad = False)
    model = Test()

    torch.onnx.export(model, image, "exports/test.onnx",
                      export_params = True,
                      opset_version = 11, 
                      do_constant_folding = True,
                      input_names = ['input'],
                      output_names = ['output']