compress_nlvr_nlvr2_p0.8.sh

#!/bin/bash
save_path=output/nlvr_nlvr2_compression_p0.8
mkdir $save_path

python -m torch.distributed.run --nproc_per_node=8 --master_port 20603 compress_nlvr_dtp.py --epoch 25 --p 0.8 \
--pretrained pretrained/model_base_nlvr.pth \
--config ./configs/nlvr.yaml \
--output_dir $save_path >$save_path/train.log