outputs/gpt2cocoSA/tokenized_data_SA/train/state.json

{
  "_data_files": [
    {
      "filename": "data-00000-of-00001.arrow"
    }
  ],
  "_fingerprint": "4dd9af3bd7a257a5",
  "_format_columns": [
    "aesthetics_score",
    "attention_mask",
    "input_ids",
    "labels",
    "prompt",
    "query",
    "similarity"
  ],
  "_format_kwargs": {},
  "_format_type": "torch",
  "_output_all_columns": false,
  "_split": null
}