launch.json 959 B

1234567891011121314151617181920212223242526
  1. {
  2. // Use IntelliSense to learn about possible attributes.
  3. // Hover to view descriptions of existing attributes.
  4. // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
  5. "version": "0.2.0",
  6. "configurations": [
  7. {
  8. "name": "Train Doc Model",
  9. "type": "python",
  10. "request": "launch",
  11. "program": "${workspaceFolder}/train_model.py",
  12. "args": [
  13. "--data-dirs", "dataset/selfCollectedData_DocCyclic", "dataset/smartdocData_DocTrainC",
  14. "--validation-dirs", "dataset/smartDocData_DocTestC",
  15. "--name", "DocModel",
  16. "--lr", "0.5",
  17. "--batch-size", "16",
  18. "--schedule", "20", "30", "35",
  19. "--model-type", "resnet",
  20. "--loader", "ram",
  21. ],
  22. "console": "integratedTerminal",
  23. "justMyCode": true
  24. }
  25. ]
  26. }