Files
enginex-bi_series-image-cla…/docker.sh
2026-04-08 06:16:35 +00:00

7 lines
406 B
Bash
Executable File

docker stop bi150_ic
docker rm bi150_ic
docker build . -t bi150_image_classification
docker run -p 17777:8000 -v /mnt/contest_ceph/aiyueqi/image_classification/microsoft/resnet-50/:/model:ro -it --device=/dev/iluvatar0:/dev/iluvatar0 --name bi150_ic -e CONFIG_JSON='{"model_class": "AutoModelForImageClassification", "processer": "AutoImageProcessor", "torch_dtype": "auto"}' bi150_image_classification