-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
the first version of VITA reproduction, but semm no valid
- Loading branch information
root
committed
Sep 19, 2022
1 parent
50c53e2
commit b135bfb
Showing
1,014 changed files
with
123,298 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,270 @@ | ||
version: 2.1 | ||
|
||
# ------------------------------------------------------------------------------------- | ||
# Environments to run the jobs in | ||
# ------------------------------------------------------------------------------------- | ||
cpu: &cpu | ||
machine: | ||
image: ubuntu-2004:202107-02 | ||
resource_class: medium | ||
|
||
gpu: &gpu | ||
machine: | ||
# NOTE: use a cuda vesion that's supported by all our pytorch versions | ||
image: ubuntu-1604-cuda-11.1:202012-01 | ||
resource_class: gpu.nvidia.small | ||
|
||
windows-cpu: &windows_cpu | ||
machine: | ||
resource_class: windows.medium | ||
image: windows-server-2019-vs2019:stable | ||
shell: powershell.exe | ||
|
||
# windows-gpu: &windows_gpu | ||
# machine: | ||
# resource_class: windows.gpu.nvidia.medium | ||
# image: windows-server-2019-nvidia:stable | ||
|
||
version_parameters: &version_parameters | ||
parameters: | ||
pytorch_version: | ||
type: string | ||
torchvision_version: | ||
type: string | ||
pytorch_index: | ||
type: string | ||
# use test wheels index to have access to RC wheels | ||
# https://download.pytorch.org/whl/test/torch_test.html | ||
default: "https://download.pytorch.org/whl/torch_stable.html" | ||
python_version: # NOTE: only affect linux | ||
type: string | ||
default: '3.7.9' | ||
|
||
environment: | ||
PYTORCH_VERSION: << parameters.pytorch_version >> | ||
TORCHVISION_VERSION: << parameters.torchvision_version >> | ||
PYTORCH_INDEX: << parameters.pytorch_index >> | ||
PYTHON_VERSION: << parameters.python_version>> | ||
# point datasets to ~/.torch so it's cached in CI | ||
DETECTRON2_DATASETS: ~/.torch/datasets | ||
|
||
# ------------------------------------------------------------------------------------- | ||
# Re-usable commands | ||
# ------------------------------------------------------------------------------------- | ||
# install_nvidia_driver: &install_nvidia_driver | ||
# - run: | ||
# name: Install nvidia driver | ||
# working_directory: ~/ | ||
# command: | | ||
# wget -q 'https://s3.amazonaws.com/ossci-linux/nvidia_driver/NVIDIA-Linux-x86_64-430.40.run' | ||
# sudo /bin/bash ./NVIDIA-Linux-x86_64-430.40.run -s --no-drm | ||
# nvidia-smi | ||
|
||
add_ssh_keys: &add_ssh_keys | ||
# https://circleci.com/docs/2.0/add-ssh-key/ | ||
- add_ssh_keys: | ||
fingerprints: | ||
- "e4:13:f2:22:d4:49:e8:e4:57:5a:ac:20:2f:3f:1f:ca" | ||
|
||
install_python: &install_python | ||
- run: | ||
name: Install Python | ||
working_directory: ~/ | ||
command: | | ||
# upgrade pyenv | ||
cd /opt/circleci/.pyenv/plugins/python-build/../.. && git pull && cd - | ||
pyenv install -s $PYTHON_VERSION | ||
pyenv global $PYTHON_VERSION | ||
python --version | ||
which python | ||
pip install --upgrade pip | ||
setup_venv: &setup_venv | ||
- run: | ||
name: Setup Virtual Env | ||
working_directory: ~/ | ||
command: | | ||
python -m venv ~/venv | ||
echo ". ~/venv/bin/activate" >> $BASH_ENV | ||
. ~/venv/bin/activate | ||
python --version | ||
which python | ||
which pip | ||
pip install --upgrade pip | ||
setup_venv_win: &setup_venv_win | ||
- run: | ||
name: Setup Virutal Env for Windows | ||
command: | | ||
pip install virtualenv | ||
python -m virtualenv env | ||
.\env\Scripts\activate | ||
python --version | ||
which python | ||
which pip | ||
install_linux_dep: &install_linux_dep | ||
- run: | ||
name: Install Dependencies | ||
command: | | ||
# disable crash coredump, so unittests fail fast | ||
sudo systemctl stop apport.service | ||
# install from github to get latest; install iopath first since fvcore depends on it | ||
pip install --progress-bar off -U 'git+https://github.com/facebookresearch/iopath' | ||
pip install --progress-bar off -U 'git+https://github.com/facebookresearch/fvcore' | ||
# Don't use pytest-xdist: cuda tests are unstable under multi-process workers. | ||
pip install --progress-bar off ninja opencv-python-headless pytest tensorboard pycocotools | ||
pip install --progress-bar off torch==$PYTORCH_VERSION -f $PYTORCH_INDEX | ||
if [[ "$TORCHVISION_VERSION" == "master" ]]; then | ||
pip install git+https://github.com/pytorch/vision.git | ||
else | ||
pip install --progress-bar off torchvision==$TORCHVISION_VERSION -f $PYTORCH_INDEX | ||
fi | ||
python -c 'import torch; print("CUDA:", torch.cuda.is_available())' | ||
gcc --version | ||
install_detectron2: &install_detectron2 | ||
- run: | ||
name: Install Detectron2 | ||
command: | | ||
# Remove first, in case it's in the CI cache | ||
pip uninstall -y detectron2 | ||
pip install --progress-bar off -e .[all] | ||
python -m detectron2.utils.collect_env | ||
./datasets/prepare_for_tests.sh | ||
run_unittests: &run_unittests | ||
- run: | ||
name: Run Unit Tests | ||
command: | | ||
pytest -v --durations=15 tests # parallel causes some random failures | ||
uninstall_tests: &uninstall_tests | ||
- run: | ||
name: Run Tests After Uninstalling | ||
command: | | ||
pip uninstall -y detectron2 | ||
# Remove built binaries | ||
rm -rf build/ detectron2/*.so | ||
# Tests that code is importable without installation | ||
PYTHONPATH=. ./.circleci/import-tests.sh | ||
# ------------------------------------------------------------------------------------- | ||
# Jobs to run | ||
# ------------------------------------------------------------------------------------- | ||
jobs: | ||
linux_cpu_tests: | ||
<<: *cpu | ||
<<: *version_parameters | ||
|
||
working_directory: ~/detectron2 | ||
|
||
steps: | ||
- checkout | ||
|
||
# Cache the venv directory that contains python, dependencies, and checkpoints | ||
# Refresh the key when dependencies should be updated (e.g. when pytorch releases) | ||
- restore_cache: | ||
keys: | ||
- cache-{{ arch }}-<< parameters.pytorch_version >>-{{ .Branch }}-20210827 | ||
|
||
- <<: *install_python | ||
- <<: *install_linux_dep | ||
- <<: *install_detectron2 | ||
- <<: *run_unittests | ||
- <<: *uninstall_tests | ||
|
||
- save_cache: | ||
paths: | ||
- /opt/circleci/.pyenv | ||
- ~/.torch | ||
key: cache-{{ arch }}-<< parameters.pytorch_version >>-{{ .Branch }}-20210827 | ||
|
||
|
||
linux_gpu_tests: | ||
<<: *gpu | ||
<<: *version_parameters | ||
|
||
working_directory: ~/detectron2 | ||
|
||
steps: | ||
- checkout | ||
|
||
- restore_cache: | ||
keys: | ||
- cache-{{ arch }}-<< parameters.pytorch_version >>-{{ .Branch }}-20210827 | ||
|
||
- <<: *install_python | ||
- <<: *install_linux_dep | ||
- <<: *install_detectron2 | ||
- <<: *run_unittests | ||
- <<: *uninstall_tests | ||
|
||
- save_cache: | ||
paths: | ||
- /opt/circleci/.pyenv | ||
- ~/.torch | ||
key: cache-{{ arch }}-<< parameters.pytorch_version >>-{{ .Branch }}-20210827 | ||
|
||
windows_cpu_build: | ||
<<: *windows_cpu | ||
<<: *version_parameters | ||
steps: | ||
- <<: *add_ssh_keys | ||
- checkout | ||
- <<: *setup_venv_win | ||
|
||
# Cache the env directory that contains dependencies | ||
- restore_cache: | ||
keys: | ||
- cache-{{ arch }}-<< parameters.pytorch_version >>-{{ .Branch }}-20210404 | ||
|
||
- run: | ||
name: Install Dependencies | ||
command: | | ||
pip install certifi --ignore-installed # required on windows to workaround some cert issue | ||
pip install numpy cython # required on windows before pycocotools | ||
pip install opencv-python-headless pytest-xdist pycocotools tensorboard | ||
pip install -U git+https://github.com/facebookresearch/iopath | ||
pip install -U git+https://github.com/facebookresearch/fvcore | ||
pip install torch==$env:PYTORCH_VERSION torchvision==$env:TORCHVISION_VERSION -f $env:PYTORCH_INDEX | ||
- save_cache: | ||
paths: | ||
- env | ||
key: cache-{{ arch }}-<< parameters.pytorch_version >>-{{ .Branch }}-20210404 | ||
|
||
- <<: *install_detectron2 | ||
# TODO: unittest fails for now | ||
|
||
workflows: | ||
version: 2 | ||
regular_test: | ||
jobs: | ||
- linux_cpu_tests: | ||
name: linux_cpu_tests_pytorch1.10 | ||
pytorch_version: '1.10.0+cpu' | ||
torchvision_version: '0.11.1+cpu' | ||
- linux_gpu_tests: | ||
name: linux_gpu_tests_pytorch1.8 | ||
pytorch_version: '1.8.1+cu111' | ||
torchvision_version: '0.9.1+cu111' | ||
- linux_gpu_tests: | ||
name: linux_gpu_tests_pytorch1.9 | ||
pytorch_version: '1.9+cu111' | ||
torchvision_version: '0.10+cu111' | ||
- linux_gpu_tests: | ||
name: linux_gpu_tests_pytorch1.10 | ||
pytorch_version: '1.10+cu111' | ||
torchvision_version: '0.11.1+cu111' | ||
- linux_gpu_tests: | ||
name: linux_gpu_tests_pytorch1.10_python39 | ||
pytorch_version: '1.10+cu111' | ||
torchvision_version: '0.11.1+cu111' | ||
python_version: '3.9.6' | ||
- windows_cpu_build: | ||
pytorch_version: '1.10+cpu' | ||
torchvision_version: '0.11.1+cpu' |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,16 @@ | ||
#!/bin/bash -e | ||
# Copyright (c) Facebook, Inc. and its affiliates. | ||
|
||
# Test that import works without building detectron2. | ||
|
||
# Check that _C is not importable | ||
python -c "from detectron2 import _C" > /dev/null 2>&1 && { | ||
echo "This test should be run without building detectron2." | ||
exit 1 | ||
} | ||
|
||
# Check that other modules are still importable, even when _C is not importable | ||
python -c "from detectron2 import modeling" | ||
python -c "from detectron2 import modeling, data" | ||
python -c "from detectron2 import evaluation, export, checkpoint" | ||
python -c "from detectron2 import utils, engine" |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,85 @@ | ||
AccessModifierOffset: -1 | ||
AlignAfterOpenBracket: AlwaysBreak | ||
AlignConsecutiveAssignments: false | ||
AlignConsecutiveDeclarations: false | ||
AlignEscapedNewlinesLeft: true | ||
AlignOperands: false | ||
AlignTrailingComments: false | ||
AllowAllParametersOfDeclarationOnNextLine: false | ||
AllowShortBlocksOnASingleLine: false | ||
AllowShortCaseLabelsOnASingleLine: false | ||
AllowShortFunctionsOnASingleLine: Empty | ||
AllowShortIfStatementsOnASingleLine: false | ||
AllowShortLoopsOnASingleLine: false | ||
AlwaysBreakAfterReturnType: None | ||
AlwaysBreakBeforeMultilineStrings: true | ||
AlwaysBreakTemplateDeclarations: true | ||
BinPackArguments: false | ||
BinPackParameters: false | ||
BraceWrapping: | ||
AfterClass: false | ||
AfterControlStatement: false | ||
AfterEnum: false | ||
AfterFunction: false | ||
AfterNamespace: false | ||
AfterObjCDeclaration: false | ||
AfterStruct: false | ||
AfterUnion: false | ||
BeforeCatch: false | ||
BeforeElse: false | ||
IndentBraces: false | ||
BreakBeforeBinaryOperators: None | ||
BreakBeforeBraces: Attach | ||
BreakBeforeTernaryOperators: true | ||
BreakConstructorInitializersBeforeComma: false | ||
BreakAfterJavaFieldAnnotations: false | ||
BreakStringLiterals: false | ||
ColumnLimit: 80 | ||
CommentPragmas: '^ IWYU pragma:' | ||
ConstructorInitializerAllOnOneLineOrOnePerLine: true | ||
ConstructorInitializerIndentWidth: 4 | ||
ContinuationIndentWidth: 4 | ||
Cpp11BracedListStyle: true | ||
DerivePointerAlignment: false | ||
DisableFormat: false | ||
ForEachMacros: [ FOR_EACH, FOR_EACH_R, FOR_EACH_RANGE, ] | ||
IncludeCategories: | ||
- Regex: '^<.*\.h(pp)?>' | ||
Priority: 1 | ||
- Regex: '^<.*' | ||
Priority: 2 | ||
- Regex: '.*' | ||
Priority: 3 | ||
IndentCaseLabels: true | ||
IndentWidth: 2 | ||
IndentWrappedFunctionNames: false | ||
KeepEmptyLinesAtTheStartOfBlocks: false | ||
MacroBlockBegin: '' | ||
MacroBlockEnd: '' | ||
MaxEmptyLinesToKeep: 1 | ||
NamespaceIndentation: None | ||
ObjCBlockIndentWidth: 2 | ||
ObjCSpaceAfterProperty: false | ||
ObjCSpaceBeforeProtocolList: false | ||
PenaltyBreakBeforeFirstCallParameter: 1 | ||
PenaltyBreakComment: 300 | ||
PenaltyBreakFirstLessLess: 120 | ||
PenaltyBreakString: 1000 | ||
PenaltyExcessCharacter: 1000000 | ||
PenaltyReturnTypeOnItsOwnLine: 200 | ||
PointerAlignment: Left | ||
ReflowComments: true | ||
SortIncludes: true | ||
SpaceAfterCStyleCast: false | ||
SpaceBeforeAssignmentOperators: true | ||
SpaceBeforeParens: ControlStatements | ||
SpaceInEmptyParentheses: false | ||
SpacesBeforeTrailingComments: 1 | ||
SpacesInAngles: false | ||
SpacesInContainerLiterals: true | ||
SpacesInCStyleCastParentheses: false | ||
SpacesInParentheses: false | ||
SpacesInSquareBrackets: false | ||
Standard: Cpp11 | ||
TabWidth: 8 | ||
UseTab: Never |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,15 @@ | ||
# This is an example .flake8 config, used when developing *Black* itself. | ||
# Keep in sync with setup.cfg which is used for source packages. | ||
|
||
[flake8] | ||
ignore = W503, E203, E221, C901, C408, E741, C407, B017, F811 | ||
max-line-length = 100 | ||
max-complexity = 18 | ||
select = B,C,E,F,W,T4,B9 | ||
exclude = build | ||
per-file-ignores = | ||
**/__init__.py:F401,F403,E402 | ||
**/configs/**.py:F401,E402 | ||
configs/**.py:F401,E402 | ||
**/tests/config/**.py:F401,E402 | ||
tests/config/**.py:F401,E402 |
Oops, something went wrong.