config.json

{
  "architectures": [
    "SimCSEModel"
  ],
  "base_model": "klue/roberta-small",
  "batch_size": 64,
  "dataset_dir": "dataset-sup-train",
  "debug": "True",
  "device": "cuda:0",
  "eval_steps": 50,
  "lr": 3e-05,
  "lr_scheduler_type": "linear",
  "max_seq_len": 32,
  "num_epochs": 3,
  "num_warmup_steps": 0,
  "output_dir": "./sup-model_klue",
  "seed": 42,
  "temperature": 0.05,
  "torch_dtype": "float32",
  "transformers_version": "4.41.2",
  "version": 1.0
}