launch.json 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  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": "convert_model_to_tflite.py",
  9. "type": "python",
  10. "request": "launch",
  11. "program": "${workspaceFolder}/convert_model_to_tflite.py",
  12. "console": "integratedTerminal",
  13. "args": ["--ckpt_path=/Users/chenlong/Desktop/models/model_ssmi_all_3.pt"],
  14. // "justMyCode": false
  15. },
  16. {
  17. "name": "tflite_infer.py",
  18. "type": "python",
  19. "request": "launch",
  20. "program": "${workspaceFolder}/tflite_infer.py",
  21. "console": "integratedTerminal"
  22. },
  23. {
  24. "name": "train",
  25. "type": "python",
  26. "request": "launch",
  27. "program": "${workspaceFolder}/train.py",
  28. "console": "integratedTerminal",
  29. "args": [
  30. "--develop=true",
  31. "--lr=1e-3",
  32. "--retrain=true",
  33. ]
  34. // "justMyCode": false
  35. },
  36. {
  37. "name": "generate_dataset",
  38. "type": "python",
  39. "request": "launch",
  40. "program": "${workspaceFolder}/generate_dataset.py",
  41. "console": "integratedTerminal",
  42. // "justMyCode": false
  43. },
  44. {
  45. "name": "infer",
  46. "type": "python",
  47. "request": "launch",
  48. "program": "${workspaceFolder}/infer.py",
  49. "args": ["--ckpt_path=/Users/chenlong/Desktop/models/model_ssmi_all_5.pt"],
  50. "console": "integratedTerminal",
  51. // "justMyCode": false
  52. }
  53. ]
  54. }