molcrawl-compounds-guacamol-gpt2-small/training_args.json

{
  "iteration": 2000,
  "best_val_loss": 0.031413231045007706,
  "early_stopping_counter": 1,
  "learning_rate": 1e-05,
  "batch_size": 8,
  "block_size": 1024,
  "model_args": {
    "n_layer": 12,
    "n_head": 12,
    "n_embd": 768,
    "block_size": 1024,
    "bias": false,
    "vocab_size": 612,
    "dropout": 0.0
  }
}