scripts/peptide_generation.sh

export CUDA_VISIBLE_DEVICES=2

python ./peptide/generation.py \
  --checkpoint ./peptide/ckpt/PepReDi_v3.pt \
  --num_samples 100 \
  --output_file ./peptide/generated_samples.txt \
  --gen_steps 16 \
  --gen_len 14 \
  --temperature 1