init project files

2026-04-08 06:16:35 +00:00
parent f8f29d178c
commit d2f4885f07
6 changed files with 161 additions and 0 deletions
--- a/docker.sh
+++ b/docker.sh
@@ -0,0 +1,6 @@
+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