wehub-resource-sync
79 lines
2.3 KiB
Bash
79 lines
2.3 KiB
Bash
#!/usr/bin/env zsh
|
|
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
#
|
|
# This source code is licensed under the MIT license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
|
|
source_dir=$1
|
|
tgt_dir=$2
|
|
model=$3
|
|
|
|
if [ -z "$4" ]
|
|
then
|
|
dim=512
|
|
else
|
|
dim=$4
|
|
fi
|
|
|
|
echo "using $dim dim for PCA"
|
|
|
|
if [ -z "$5" ]
|
|
then
|
|
layer=14
|
|
else
|
|
layer=$5
|
|
fi
|
|
|
|
echo "extracting from layer $layer"
|
|
|
|
train_split=train
|
|
valid_split=valid
|
|
test_split=test
|
|
|
|
all_splits=($train_split)
|
|
|
|
if [[ -f "$source_dir/valid.tsv" ]]; then
|
|
all_splits+=('valid')
|
|
fi
|
|
|
|
if [[ -f "$source_dir/test.tsv" ]]; then
|
|
all_splits+=('test')
|
|
fi
|
|
|
|
echo "processing splits: $all_splits"
|
|
|
|
mkdir -p $tgt_dir
|
|
|
|
cp $source_dir/*.tsv $tgt_dir
|
|
cp $source_dir/*.wrd $tgt_dir
|
|
cp $source_dir/*.ltr $tgt_dir
|
|
cp $source_dir/*.phn $tgt_dir
|
|
cp $source_dir/dict* $tgt_dir
|
|
|
|
setopt shwordsplit
|
|
|
|
for split in $all_splits; do
|
|
python $FAIRSEQ_ROOT/examples/wav2vec/unsupervised/scripts/wav2vec_extract_features.py $source_dir --split $split \
|
|
--save-dir $tgt_dir --checkpoint $model --layer $layer
|
|
done
|
|
|
|
python $FAIRSEQ_ROOT/examples/wav2vec/unsupervised/scripts/wav2vec_cluster_faiss.py $tgt_dir/${train_split}.tsv \
|
|
--checkpoint $model --save-dir $tgt_dir -f "CLUS128" --sample-pct 1.0
|
|
|
|
for split in $all_splits; do
|
|
python $FAIRSEQ_ROOT/examples/wav2vec/unsupervised/scripts/wav2vec_apply_cluster_faiss.py $tgt_dir \
|
|
--checkpoint $model --path $tgt_dir/CLUS128 --split $split
|
|
done
|
|
|
|
python $FAIRSEQ_ROOT/examples/wav2vec/unsupervised/scripts/pca.py $tgt_dir/${train_split}.npy --output $tgt_dir/pca --dim $dim
|
|
|
|
for split in $all_splits; do
|
|
python $FAIRSEQ_ROOT/examples/wav2vec/unsupervised/scripts/apply_pca.py $tgt_dir --split $split --save-dir $tgt_dir/precompute_pca$dim --pca-path $tgt_dir/pca/${dim}_pca --batch-size 1048000
|
|
|
|
python $FAIRSEQ_ROOT/examples/wav2vec/unsupervised/scripts/merge_clusters.py $tgt_dir/precompute_pca$dim --cluster-dir $tgt_dir/CLUS128 \
|
|
--split $split --save-dir $tgt_dir/precompute_pca${dim}_cls128_mean --pooling mean
|
|
|
|
python $FAIRSEQ_ROOT/examples/wav2vec/unsupervised/scripts/mean_pool.py $tgt_dir/precompute_pca${dim}_cls128_mean \
|
|
--save-dir $tgt_dir/precompute_pca${dim}_cls128_mean_pooled --split $split
|
|
done
|