config.json

{
  "device": 0,
  "drop_ratio": 0.25,
  "num_layer": 5,
  "hidden_size": 768,
  "temperature": 1,
  "freeze_text_encoder": true,
  "batch_size": 1000,
  "epochs": 40,
  "patience": 10,
  "lr": 0.001,
  "l2reg": 5e-06
}