d2693e0
1
2
3
4
5
6
7
8
9
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
