Spaces:
Sleeping
Sleeping
Update compute_model_property.py
Browse files
compute_model_property.py
CHANGED
@@ -82,10 +82,10 @@ def return_output(model_checkpoint, dataset_name):
|
|
82 |
"""
|
83 |
Please enter model and dataset identifiers
|
84 |
"""
|
85 |
-
model_checkpoint = "sgugger/glue-mrpc"
|
86 |
-
dataset_name = "nyu-mll/glue"
|
87 |
|
88 |
-
print(model_checkpoint, dataset_name, metric
|
89 |
|
90 |
model_checkpoint = model_checkpoint
|
91 |
raw_datasets = load_dataset(dataset_name, "mrpc")
|
|
|
82 |
"""
|
83 |
Please enter model and dataset identifiers
|
84 |
"""
|
85 |
+
#model_checkpoint = "sgugger/glue-mrpc"
|
86 |
+
#dataset_name = "nyu-mll/glue"
|
87 |
|
88 |
+
print(model_checkpoint, dataset_name, metric)
|
89 |
|
90 |
model_checkpoint = model_checkpoint
|
91 |
raw_datasets = load_dataset(dataset_name, "mrpc")
|