-
Notifications
You must be signed in to change notification settings - Fork 4
Conversation
trainer.sh takes in github url, oauth token as argument and clones a private repo Dockerfile is based on Python 3.7 base image, and calls the trainer script
Addresses part of #5 |
docker/trainer.sh
Outdated
r) git_url=${OPTARG};; | ||
t) oauth_token=${OPTARG};; | ||
p) project_name=${OPTARG}; | ||
esac |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Question - are these options being read as environment variables ? for eg git_url should be read from $GIT_URL
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Ok fixed!
Ubuntu has better compatibility with data scientists
docker/trainer.sh
Outdated
|
||
cd $project_name | ||
pip install -r requirements.txt | ||
python train.py |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Lets now take the model file to run as a relative path
via environment variable $MODEL_PATH
docker/Dockerfile
Outdated
@@ -0,0 +1,6 @@ | |||
FROM python:3.7-buster |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
can you use ubuntu as base image ?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Also any specific reason for using 3.7-buster
?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Hey I changed the base image to Ubuntu 20.04
trainer.sh takes in github url, oauth token as argument and clones a private repo
Dockerfile is based on Python 3.7 base image, and calls the trainer script