Commit 9846958a authored by zhanggzh's avatar zhanggzh
Browse files

增加keras-cv模型及训练代码

parent 17bc28d5
ARG VERSION
FROM tensorflow/tensorflow:${VERSION}
RUN apt-get update && apt-get install -y \
locales \
&& rm -rf /var/lib/apt/lists/*
RUN pip install \
black \
flake8 \
isort \
pytest \
regex \
tensorflow_datasets
{
"name": "Keras-cv",
"build": {
"dockerfile": "Dockerfile",
"args": {
"VERSION": "2.10.0"
// Uncomment this if GPU support is required
// "VERSION": "2.10.0-gpu",
}
},
"settings": {
"python.pythonPath": "/usr/bin/python3",
"python.linting.enabled": true,
"python.linting.flake8Enabled": true,
"python.testing.pytestEnabled":true,
"python.editor.defaultFormatter": "ms-python.black-formatter",
"python.editor.formatOnSave": true,
"python.editor.codeActionsOnSave": {
"source.organizeImports": true
}
},
"extensions": [
"ms-python.python",
"ms-python.isort",
"ms-python.black-formatter"
],
"features": {
"git": {
"version": "os-provided"
}
},
// TODO: Improve to allow dynamic runArgs, see microsoft/vscode-remote-release#3972
// Uncomment this if GPU support is required
// "runArgs": [
// "--gpus=all"
// ],
"onCreateCommand": "locale-gen \"en_US.UTF-8\""
// Optional: install pre-commit hooks
// "postCreateCommand": "git config core.hooksPath .github/.githooks"
}
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment