REVA-QCAV/.vscode/launch.json
Laurent Fainsin c312513eff feat: checkpointing + torchmetrics + ...
Former-commit-id: 781a11f646964567b3bc6831caa5e380748b84e1 [formerly 7ea5f37ecb7fd5c95e5d1b7b1aa899d986ecff2a]
Former-commit-id: 06974fa163a6a8eb881d5981f9c6debe63f1b4bb
2022-09-05 16:04:52 +02:00

30 lines
637 B
JSON

{
"version": "0.2.0",
"configurations": [
{
"name": "Train",
"type": "python",
"request": "launch",
"program": "${workspaceFolder}/src/train.py",
"console": "integratedTerminal",
"justMyCode": false,
},
{
"name": "Predict",
"type": "python",
"request": "launch",
"program": "${workspaceFolder}/src/predict.py",
"console": "integratedTerminal",
"justMyCode": false,
"args": [
"--input",
"images/input.png",
"--output",
"images/output.png",
"--model",
"checkpoints/model.onnx"
]
}
]
}