Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

dvc.yaml 2.3 KB

You have to be logged in to leave a comment. Sign In
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
  1. stages:
  2. train-chess:
  3. desc: Train PoseNet on chess scene
  4. cmd: python train.py --config configs/pose_net.yaml --dataset_name chess
  5. --out models/chess_pose_net.pth --seed 1 --gpu 1 --log_every_n_steps 1
  6. deps:
  7. - train.py
  8. - lieposenet
  9. outs:
  10. - models/chess_pose_net.pth.ckpt
  11. params:
  12. - configs/pose_net.yaml:
  13. - model
  14. - max_epochs
  15. - data_module
  16. test-chess:
  17. desc: Test Pose Net
  18. cmd: python eval.py
  19. --config configs/pose_net.yaml --dataset_name chess --result metrics/chess_metrics.json
  20. --data_saver_path metrics/chess_trajectories.npy --model models/chess_pose_net.pth.ckpt --seed 1 --gpu 1
  21. --log_every_n_steps 1
  22. deps:
  23. - eval.py
  24. - lieposenet
  25. - models/chess_pose_net.pth.ckpt
  26. params:
  27. - configs/pose_net.yaml:
  28. - model
  29. - data_module
  30. outs:
  31. - metrics/chess_trajectories.npy
  32. metrics:
  33. - metrics/chess_metrics.json:
  34. cache: false
  35. train-fire:
  36. desc: Train PoseNet on fire scene
  37. cmd: python train.py --config configs/pose_net.yaml --dataset_name fire
  38. --out models/fire_pose_net.pth --seed 1 --gpu 1 --log_every_n_steps 1
  39. deps:
  40. - train.py
  41. - lieposenet
  42. outs:
  43. - models/fire_pose_net.pth.ckpt
  44. params:
  45. - configs/pose_net.yaml:
  46. - model
  47. - max_epochs
  48. - data_module
  49. test-fire:
  50. desc: Test Pose Net
  51. cmd: python eval.py
  52. --config configs/pose_net.yaml --dataset_name fire --result metrics/fire_metrics.json
  53. --data_saver_path metrics/fire_trajectories.npy --model models/fire_pose_net.pth.ckpt --seed 1 --gpu 1
  54. --log_every_n_steps 1
  55. deps:
  56. - eval.py
  57. - lieposenet
  58. - models/fire_pose_net.pth.ckpt
  59. params:
  60. - configs/pose_net.yaml:
  61. - model
  62. - data_module
  63. outs:
  64. - metrics/fire_trajectories.npy
  65. metrics:
  66. - metrics/fire_metrics.json:
  67. cache: false
Tip!

Press p or to see the previous file or, n or to see the next file

Comments

Loading...