{
    "batch_size": 128,
    "num_epochs": 1,
    "max_steps": 300000,
    "learning_rate": 3e-4,
    "weight_decay": 1e-4,
    "seed": 0
}