File size: 710 Bytes
ca32d55
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
28
29
30
31
32
33
34
#!/bin/bash

for d in work/*
do
    echo $d
    pushd $d >/dev/null

    if [ -d encoder ]; then
        xcrun coremlcompiler compile encoder/chunked_pipeline.mlpackage .
        rm -rf encoder.mlmodelc
        mv chunked_pipeline.mlmodelc encoder.mlmodelc
    else
        xcrun coremlcompiler compile encoder.mlpackage .
    fi
    xcrun coremlcompiler compile decoder_first.mlpackage .
    xcrun coremlcompiler compile decoder_second.mlpackage .

    popd >/dev/null
done

mkdir -p output
for d in work/*
do
    out=${d/work/output}
    mkdir -p $out
    mv $d/*.mlmodelc $d/model_dims.json $out/
done

mkdir -p index
for d in output/*
do
    model=${d##*/}
    (cd $d && find * -type f) > index/$model
done