FS-TFP/federatedscope/nlp/hetero_tasks/run_atc.sh

15 lines
344 B
Bash

set -e
cd ../../..
DEVICE=$1
DEBUG=False
python federatedscope/main.py \
--cfg federatedscope/nlp/hetero_tasks/baseline/config_atc.yaml \
--client_cfg federatedscope/nlp/hetero_tasks/baseline/config_client_atc.yaml \
federate.atc_load_from exp/atc/pretrain/ckpt/ \
outdir exp/atc/train/ \
device $DEVICE \
data.is_debug $DEBUG \