train.sh

export CUDA_VISIBLE_DEVICES=2,3
    ##--fsdp "full_shard auto_wrap" --fsdp_transformer_layer_cls_to_wrap 'OPTDecoderLayer' \
torchrun --nproc_per_node=2 --master_port=8081 train.py \
    --data_path ../downstream_data/gene_ranking_20220803.txt \
    --bf16 True \
    --output_dir checkpoint \
    --num_train_epochs 40 \
    --per_device_train_batch_size 512 \
    --per_device_eval_batch_size 4 \
    --gradient_accumulation_steps 2 \
    --evaluation_strategy "no" \
    --save_strategy "steps" \
    --save_steps 2000 \
    --save_total_limit 1 \
    --learning_rate 3e-4 \
    --weight_decay 0.0 \
    --warmup_ratio 0.03 \
    --adam_beta1 0.90 \
    --adam_beta2 0.95 \
    --lr_scheduler_type "cosine" \
    --logging_steps 2 \
    --report_to tensorboard \
    --tf32 True \
    --dataloader_num_workers 1 \
    --dataloader_persistent_workers True

#--data_path ../downstream_data/val.txt \