Alec commited on
Commit
284240f
1 Parent(s): 32c7190

fix n_valid default

Browse files
Files changed (1) hide show
  1. baseline.py +1 -1
baseline.py CHANGED
@@ -25,7 +25,7 @@ def load_split(data_dir, source, split, n=np.inf):
25
  labels = [0]*len(webtext)+[1]*len(gen)
26
  return texts, labels
27
 
28
- def main(data_dir, log_dir, source='xl-1542M-k40', n_train=500000, n_valid=1000, n_jobs=-1, verbose=False):
29
  train_texts, train_labels = load_split(data_dir, source, 'train', n=n_train)
30
  valid_texts, valid_labels = load_split(data_dir, source, 'valid', n=n_valid)
31
  test_texts, test_labels = load_split(data_dir, source, 'test')
 
25
  labels = [0]*len(webtext)+[1]*len(gen)
26
  return texts, labels
27
 
28
+ def main(data_dir, log_dir, source='xl-1542M-k40', n_train=500000, n_valid=10000, n_jobs=None, verbose=False):
29
  train_texts, train_labels = load_split(data_dir, source, 'train', n=n_train)
30
  valid_texts, valid_labels = load_split(data_dir, source, 'valid', n=n_valid)
31
  test_texts, test_labels = load_split(data_dir, source, 'test')