Class Model

benchmark/python/src/models.py:17–25

class Model:
    name = None
    torch_model = None
    input_shape = []

    def __init__(self, name, torch_model, input_shape):
        self.name = name
        self.torch_model = torch_model
        self.input_shape = input_shape


class AllModel: