launch.json

{
    // Use IntelliSense to learn about possible attributes.
    // Hover to view descriptions of existing attributes.
    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
    "version": "0.2.0",
    "configurations": [
        {
            "name": "Python: Current File",
            "type": "python",
            "request": "launch",
            "program": "${file}",
            "console": "integratedTerminal",
            "justMyCode": true
        },
        {
            "name": "train.py",
            "type": "python",
            "request": "launch",
            "program": "src/train.py",
            "console": "integratedTerminal",
            "justMyCode": true,
            "args": [
                "base=configs/train.yaml",
                "trainer.num_nodes=1",
                "trainer.devices=1",
                "data.train_params.batch_size=256",
                "model.optimizer_config.lr=1e-3"
            ]
        }
    ]
}