-
Notifications
You must be signed in to change notification settings - Fork 0
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[0.3.0] Abstracting Pytorch example #24
Conversation
…lab/vscode dockerfiles
…ce is not part of the problem template list
…o post gen hook to remove unneeded banners
…model checkpoint location
…ml to problem-templates/cv
… os.path.join use
Problem with the gpu (model training) Dockerfile such that pytorch is forced to run cpu only due to the dependencies. Would need to find check if nvidia image base is needed by installing the gpu dependencies onto the current cpu image (to reduce image size). Concern is that CUDA components might be installed twice in anaconda namespace, thus increasing the image size for the gpu (model training) image. |
So installing |
Closes #17.
This is a big merge before freezing the code to 0.3.0 with a number of changes on top of abstracting the Pytorch example and creating a package-agnostic base template to implement models of the user's choosing.
Changes made: