model_type: 'linear'
device: 'cpu'