example-get-started-experiments: use cpu-friendly arch

iterative / example-repos-dev

Source code and generator scripts for example DVC projects

https://dvc.org/doc

21 stars 13 forks source link

example-get-started-experiments: use cpu-friendly arch #196

Closed dberenbaum closed 1 year ago

dberenbaum commented 1 year ago

Changed model arch to something that runs in reasonable time on cpu since I think it's not feasible for everyone trying it out to have access to gpu (or they may have some restricted access to one that they don't want to use to demo a new tool). Performance is not quite as strong but it's not a dramatic dropoff:

daavoo commented 1 year ago

BTW what is the runtime on your CPU?

dberenbaum commented 1 year ago

@dberenbaum looks like the notebook needs some cleanup, the one pushed here contains errors, commented code, etc

Sorry, fixed now. Thanks for checking.

Which is the architecture we should hardcode in the notebook?

shufflenet_v2_x2_0 (you should also see it in params.yaml).

BTW what is the runtime on your CPU?

On my CPU, a single training takes <1 minute.

daavoo commented 1 year ago

Will build and push