Skip to content

vit_h batch size 1 and batch size 8 results #354

vit_h batch size 1 and batch size 8 results

vit_h batch size 1 and batch size 8 results #354

Workflow file for this run

name: Test Installation
on:
push:
branches:
- main
pull_request:
branches:
- main
jobs:
test-installation:
runs-on: ubuntu-latest
steps:
- name: Checkout code
uses: actions/checkout@v2
- name: Set up Python 3.8
uses: actions/setup-python@v2
with:
python-version: 3.8
- name: Install dependencies
run: |
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121
pip install -e .
- name: Test import
run: |
python -c "from segment_anything_fast import dynamic_quant, sam_model_registry"