run.json

{
  "args": {
    "architecture": "pipeline",
    "model": "cross-encoder/ettin-reranker-150m-v1",
    "max_length": 7999,
    "packing_version": 2,
    "packing": "shared-state",
    "question_reads_options": true,
    "state_reads_questions": true,
    "source_checkpoint": "/Users/leejongmin/code/DecisionTransformer/checkpoints/pick-150m-v5/v5_stage2_best.pt",
    "source_step": 75
  }
}