Skip to content

Try add depth cov loss to improve quality #229

Try add depth cov loss to improve quality

Try add depth cov loss to improve quality #229

Workflow file for this run

name: Running experiment
on:
pull_request:
types: [ labeled ]
jobs:
check-label-prefix:
runs-on: ubuntu-latest
env:
LABEL_NAME: ${{ github.event.label.name }}
outputs:
need_experiment: ${{ steps.check.outputs.need_experiment }}
label_name: ${{ env.LABEL_NAME }}
steps:
- name: Checkout code
uses: actions/checkout@v3
- id: check
run: |
LABEL_NAME=$LABEL_NAME
PREFIX="need_experiment"
if [[ "$LABEL_NAME" == "$PREFIX"* ]]; then
echo "need_experiment=true"
echo "need_experiment=true" >> $GITHUB_OUTPUT
else
echo "need_experiment=false" >> $GITHUB_OUTPUT
fi
build:
needs: check-label-prefix
if: needs.check-label-prefix.outputs.need_experiment == 'true'
name: Run experiment
runs-on: ubuntu-latest
permissions:
contents: read
pull-requests: write
issues: write
steps:
- name: Check out code
uses: actions/checkout@v3
- name: Get the branch name
id: get-branch-name
run: echo "::set-output name=BRANCH_NAME::$(echo ${GITHUB_HEAD_REF:-${GITHUB_REF#refs/heads/}})"
- name: Configure AWS credentials
uses: aws-actions/configure-aws-credentials@v1
with:
aws-access-key-id: ${{ secrets.AWS_ACCESS_KEY_ID }}
aws-secret-access-key: ${{ secrets.AWS_SECRET_ACCESS_KEY }}
aws-region: us-east-2
- name: Login to Amazon ECR
id: login-ecr
uses: aws-actions/amazon-ecr-login@v1
- name: Build, tag, and push image to Amazon ECR
id: build-image
env:
ECR_REGISTRY: ${{ steps.login-ecr.outputs.registry }}
ECR_REPOSITORY: taichi_3d_gaussian_splatting
# use the branch name for the Docker tag
IMAGE_TAG: ${{ steps.get-branch-name.outputs.BRANCH_NAME }}
run: |
docker system prune -af
docker volume prune -f
aws ecr get-login-password --region us-east-2 | docker login --username AWS --password-stdin 763104351884.dkr.ecr.us-east-2.amazonaws.com
docker build -t $ECR_REGISTRY/$ECR_REPOSITORY:$IMAGE_TAG -f Dockerfile.aws .
docker push $ECR_REGISTRY/$ECR_REPOSITORY:$IMAGE_TAG
echo "::set-output name=image_uri::$ECR_REGISTRY/$ECR_REPOSITORY:$IMAGE_TAG"
- name: Set up Python
uses: actions/setup-python@v2
with:
python-version: '3.10'
cache: 'pip'
- run: pip3 install sagemaker argparse PyGithub boto3 sagemaker-experiments
- name: Run the experiment
id: run-experiment
env:
GITHUB_OWNER: ${{ github.repository_owner }}
GITHUB_REPO: ${{ github.repository }}
GITHUB_SHA: ${{ github.sha }}
BRANCH_NAME: ${{ steps.get-branch-name.outputs.BRANCH_NAME }}
GITHUB_ACCESS_TOKEN: ${{ secrets.GITHUB_TOKEN }}
PULL_REQUEST_NUMBER: ${{ github.event.pull_request.number }}
IMAGE_URI: ${{ steps.build-image.outputs.image_uri }}
LABEL_NAME: ${{ needs.check-label-prefix.outputs.label_name }}
run: |
python3 ci/run_experiment.py