{
  "input_shape": [
    2,
    8,
    400
  ],
  "token_shape": [
    2,
    8,
    20,
    128
  ],
  "masked_tokens": [
    80,
    80
  ],
  "relation_shape": [
    2,
    28,
    4
  ],
  "logit_shape": [
    2,
    2
  ],
  "device": "cpu",
  "amp_enabled": false,
  "teacher_has_gradient": false,
  "pair_symmetry_max_error": 0.0,
  "loss": 1.3752905130386353,
  "encoder_parameters": 1594496,
  "pretrainer_parameters": 3290904,
  "classifier_parameters": 1678211,
  "status": "passed"
}