tess-2-demo / shell_scripts /perplexity.sh
hamishivi's picture
commit
17ff0d8 verified
raw
history blame contribute delete
514 Bytes
python -m scripts.calculate_model_output_perplexity \
--model_name_or_path pretraining_exps/random-34k \
--simplex_value 5 \
--num_diffusion_steps 5000 \
--num_inference_diffusion_steps 100 \
--beta_schedule squaredcos_improved_ddpm \
--top_p 0.99 \
--self_condition logits_mean \
--self_condition_mix_before_weights \
--is_causal false \
--mask_padding_in_loss false \
--dataset_name c4 --streaming --dataset_config_name en