finetune_demo_msa.sh

# Copyright 2024 ByteDance and/or its affiliates.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

#export LAYERNORM_TYPE=fast_layernorm
#export USE_DEEPSPEED_EVO_ATTENTION=true
# wget -P /af3-dev/release_model/ https://af3-dev.tos-cn-beijing.volces.com/release_model/model_v0.2.0.pt
checkpoint_path="/home/lhw/work/rna2025/release_data/checkpoint//model_v0.2.0.pt"

CUDA_VISIBLE_DEVICES=1 python3 ./runner/train.py \
--run_name output_comp_with_msa \
--seed 42 \
--base_dir ./output_comp_with_msa \
--dtype bf16 \
--use_msa true \
--project protenix \
--use_wandb true \
--diffusion_batch_size 8 \
--eval_interval 50000 \
--log_interval 1 \
--checkpoint_interval 2000 \
--ema_decay 0.995 \
--train_crop_size 416 \
--max_steps 4000 \
--warmup_steps 50 \
--lr 0.0001 \
--sample_diffusion.N_step 20 \
--load_checkpoint_path ${checkpoint_path} \
--load_ema_checkpoint_path ${checkpoint_path} \
--data.train_sets weightedPDB_before2109_wopb_nometalc_0925 \
--data.weightedPDB_before2109_wopb_nometalc_0925.base_info.pdb_list examples/finetune_subset.txt \
--data.test_sets recentPDB_1536_sample384_0925,posebusters_0925

#sh infer.sh