configs/train/custom.json

{"batch_size": 8, "num_epochs": 1, "max_steps": 300, "learning_rate": 0.001, "weight_decay": 0.0001, "seed": 0}