opt_350m_gated_attention.sh

#!/bin/bash
accelerate launch --config_file accelerate_configs/1gpu_fp16.yaml run_clm.py \
--pad_to_max_length \
--with_tracking \
--report_to tensorboard \
--seed 1000 \
--dataset_setup bookcorpus_and_wiki \
--preprocessing_num_workers 4 \
--data_cache_dir ~/.hf_data \
--model_cache_dir ~/.hf_cache \
--model_type opt \
--tokenizer_name facebook/opt-350m \
--max_seq_length 2048 \
--block_size 512 \
--learning_rate 0.0004 \
--lr_scheduler_type linear \
--max_train_steps 100000 \
--num_warmup_steps 2000 \
--per_device_train_batch_size 24 \
--per_device_eval_batch_size 24 \
--gradient_accumulation_steps 11 \
--max_grad_norm 1.0 \
--weight_decay 0.1 \
--checkpointing_steps 1000 \
--config_path model_configs/opt-350m.yaml \
--attn_gate_type conditional_per_token \
--attn_gate_init 0.25 \
--output_dir output