update README
This commit is contained in:
@@ -3,4 +3,4 @@ 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
|
||||
docker run -p 17777:8000 -v /$mountpath/:/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
|
||||
|
||||
Reference in New Issue
Block a user