10 lines
746 B
Bash
10 lines
746 B
Bash
#!/bin/bash
|
|
# Collect detector spike counts from KNP for readout training: run_counts.sh <model_dir> <out_dir> [chunk]
|
|
M=$1; O=$2; C=${3:-1000}
|
|
D=/home/int/run/cifar/vehicles
|
|
export LD_LIBRARY_PATH=$HOME/knp-build/bin:$HOME/deps/lib
|
|
one() { f=$1; first=$2; n=$3; tag=$4; [ -f $O/${tag}_$first.done ] && return
|
|
~/run/altai-objects/altai_infer $D/${f}_images.bin $first $n $M $HOME/knp-build/bin/libknp-cpu-single-threaded-backend.so $O/${tag}_$first > $O/${tag}_$first.log 2>&1 && touch $O/${tag}_$first.done; }
|
|
export -f one; export M O D
|
|
{ for s in train train_flip; do for i in $(seq 0 $C 15999); do echo "$s $i $C $s"; done; done
|
|
for i in $(seq 0 $C 3999); do echo "val $i $C val"; done; } | xargs -P ${JOBS:-14} -L 1 bash -c 'one "$@"' _
|