multi-task-person-perception/time_inference.sh at master · TUI-NICR/multi-task-person-perception · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
#!/usr/bin/env bash

for BATCHSIZE in 1 5 10 20 50
do
    echo $BATCHSIZE
    # cpu
    python3 time_inference.py --model beyer_mod_relu_depth_126_48 --device cpu --batch_size $BATCHSIZE
    python3 time_inference.py --model mobilenetv2_small_additional_dense_pretrained --device cpu --batch_size $BATCHSIZE
    python3 time_inference.py --model mobilenetv2_large_additional_dense_pretrained --device cpu --batch_size $BATCHSIZE
    python3 time_inference.py --model efficientnet_b0_additional_dense_pretrained --device cpu --batch_size $BATCHSIZE
    python3 time_inference.py --model efficientnet_b1_additional_dense_pretrained --device cpu --batch_size $BATCHSIZE
    python3 time_inference.py --model resnet18_additional_dense_pretrained --device cpu --batch_size $BATCHSIZE
    python3 time_inference.py --model resnet34_additional_dense_pretrained --device cpu --batch_size $BATCHSIZE
    python3 time_inference.py --model resnet50_additional_dense_pretrained --device cpu --batch_size $BATCHSIZE
    python3 time_inference.py --model resnext50_additional_dense_pretrained --device cpu --batch_size $BATCHSIZE

    # gpu
    python3 time_inference.py --model beyer_mod_relu_depth_126_48 --floatx 32 16 --batch_size $BATCHSIZE
    python3 time_inference.py --model mobilenetv2_small_additional_dense_pretrained --floatx 32 16 --batch_size $BATCHSIZE
    python3 time_inference.py --model mobilenetv2_large_additional_dense_pretrained --floatx 32 16 --batch_size $BATCHSIZE
    python3 time_inference.py --model efficientnet_b0_additional_dense_pretrained --floatx 32 16 --batch_size $BATCHSIZE
    python3 time_inference.py --model efficientnet_b1_additional_dense_pretrained --floatx 32 16 --batch_size $BATCHSIZE
    python3 time_inference.py --model resnet18_additional_dense_pretrained --floatx 32 16 --batch_size $BATCHSIZE
    python3 time_inference.py --model resnet34_additional_dense_pretrained --floatx 32 16 --batch_size $BATCHSIZE
    python3 time_inference.py --model resnet50_additional_dense_pretrained --floatx 32 16 --batch_size $BATCHSIZE
    python3 time_inference.py --model resnext50_additional_dense_pretrained --floatx 32 16 --batch_size $BATCHSIZE
done