cdw

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from typing import List, Union
import numpy as np


class TicTacToe:
    """Play a game of tictactoe, optionally against a human"""

cdw

torch.onnx.export(model = torch_model,
                 args = (dummy_input['input_ids'].to(device), dummy_input['attention_mask'].to(device)), #Tuple of inputs for models that take more than one input
                 f = onnx_path,
                 export_params = True,
                 opset_version=11,
                 do_constant_folding = True,
                 input_names = ['input_ids', 'attention_mask'],
                 output_names = ['output'],
                 dynamic_axes = {'input_ids': {0: 'batch_size'},
                                'attention_mask': {0: 'batch_size'},

cdw

cdw