{ "image_size": 224, "batch_size": 512, "workers": 8, "d_model": 256, "n_layers": 6, "epochs": 200, "lr": 0.0003, "resume_checkpoint": null, "device": "cuda", "num_parameters": 9834504 }