ksatzke commited on
Commit
3f764f8
·
verified ·
1 Parent(s): 3485f9a

Update compute_model_property.py

Browse files
Files changed (1) hide show
  1. compute_model_property.py +3 -3
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, name)
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")