#!/bin/bash

P_BASE_VARS=0.10 0.20 0.30 0.35 0.40 0.45 0.5 0.55 0.6 0.65 0.7 0.75 0.8 0.85 0.9 0.95 1.0
K_BASE_VARS=10 30 50 100 250 500 1500 3000 8000
KTEMP_VARS=0.9 0.8 0.7 0.6

LAGS=1 2 3 4
LINEAR_BASES=2 3 4 5

K_RANDOMNESS_VARS=1 2 5 10 20 30 40 50 60 70 80 90 100 110 120 130 140 150 160 170 180 190 200 300 500 1000 2000 5000 10000
P_RANDOMNESS_VARS=0.01 0.02 0.05 0.1 0.15 0.2 0.25 0.3 0.35 0.4 0.45 0.5 0.6 0.7 0.8 0.9 0.95
P_RANDOMNESS_BASES=0.01 0.02 0.05 0.1 0.15 0.2 0.25 0.3 0.35 0.4 0.45 0.5 0.6 0.7 0.8

TEMPERATURE_VARS=0.01 0.05 0.1 0.3 0.5 0.6 0.7 0.8 0.85 0.9 0.95 1.0 1.05 1.1 1.15 1.20 1.25 1.30 1.35 1.40 1.50 1.60 1.70 1.8 1.9 2.0 3.0 4.0 5.0 8.0 10.0 15.0 20.0 30.0 50.0 100.0

K_TRUNCATION_VALUES=0 1 2 5 10 20
KNN_RANGES=2 3 4 5 6 10 20 30 40 50 70 90 100 150 200

N_GPUS=50

PRETRAINED_CLASS=models/gigaword_gpt2

GENERATION_BATCH_SIZE=80
ENCODING_BATCH_SIZE=1

PREFIX_FILE=data/gigaword/test.txt_filtered
PREFIX_LENGTH=10
GRAM=3
CTRL_CODE=Wikipedia
MAX_SEQ_LENGTH=100
NUM_SENTENCES=6000
SEED=10
EVAL_TEXT=data/gigaword/valid.txt_filtered
RESULTS_FILE=results_gigaword_amt_refreshed_num$$NUM_SENTENCES.json
EVAL_METHOD=BLEU
KNN=7
PLOT_GOLD=0
GPU_IDX=0
