File tree 1 file changed +2
-2
lines changed
pytorch/serving/model-archive
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 28
28
29
29
if [[ " $1 " == " cpu" ]]; then
30
30
wget https://download.pytorch.org/models/squeezenet1_1-b8a52dc0.pth
31
- torch-model-archiver --model-name squeezenet1_1 --version 1.1 --model-file /home/model-server/model-archive/model.py --serialized-file squeezenet1_1-b8a52dc0.pth --handler image_classifier --export-path /home/model-server/model-store
31
+ torch-model-archiver --model-name squeezenet1_1 --version 1.1 --model-file /home/model-server/model-archive/model.py --serialized-file squeezenet1_1-b8a52dc0.pth --handler image_classifier --export-path /home/model-server/model-store --force
32
32
rm -rf squeezenet1_1-b8a52dc0.pth
33
33
elif [[ " $1 " == " xpu" ]]; then
34
34
python /home/model-server/model-archive/ipex_squeezenet.py
35
- torch-model-archiver --model-name squeezenet1_1 --version 1.1 --serialized-file squeezenet1_1-jit.pt --handler image_classifier --export-path /home/model-server/model-store
35
+ torch-model-archiver --model-name squeezenet1_1 --version 1.1 --serialized-file squeezenet1_1-jit.pt --handler image_classifier --export-path /home/model-server/model-store --force
36
36
rm -rf squeezenet1_1-jit.pt
37
37
else
38
38
echo " Only cpu and xpu devices supported"
You can’t perform that action at this time.
0 commit comments