Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Alec
commited on
Commit
•
284240f
1
Parent(s):
32c7190
fix n_valid default
Browse files- 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=
|
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')
|