configs/train/custom.json

{
    "batch_size": 1024,
    "num_epochs": 1,
    "max_steps": 100000,
    "learning_rate": 3e-4,
    "weight_decay": 1e-4,
    "seed": null
}