-
Notifications
You must be signed in to change notification settings - Fork 455
/
generate_from_pretrain.sh
32 lines (30 loc) · 1.24 KB
/
generate_from_pretrain.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#!/bin/bash
python generate.py \
--device '0' \
--length 512 \
--batch_size 1 \
--nsamples 2 \
--temperature 1 \
--topk 8 \
--topp 0 \
--repetition_penalty 1. \
--save_samples \
--save_samples_dir 'samples_save_dir' \
--samples_sign 'sample_name' \
--model_dir 'model/deeprapper-model' \
--model_config 'config.json' \
--enable_final \
--enable_sentence \
--enable_relative_pos \
--enable_beat \
--prefix '我长大的地方像一个简朴的寨' \
--with_beat \
--beat_mode 0 \
--reverse \
--pattern 'beam' \
--beam_sample_select_sg 'sample' \
--beam_cut_temperature 10 \
--dynamic_rhyme \
--rhyme_count 1 \
--rhyme_prob_bound 1.0 \
--rhyme_alpha 0.95