{ | |
"attribute_encoder_type": "text", | |
"continuous_head": "mse", | |
"exog_input_dim": 3, | |
"ignore_attributes": false, | |
"mlp_hidden_dim": 256, | |
"onehot_attr_input_dim": 187, | |
"qoi_dim": 1, | |
"resnet_hidden_dim": 1024, | |
"resnet_num_blocks": 8, | |
"text_encoder_name": "distilbert-base-uncased", | |
"text_finetune_only_specific_layers": true, | |
"text_freeze_encoder": false | |
} |