feat: add model name to results

wojtek11530 · wojtek11530 · commit f8df472777d9 · 2021-12-17T08:41:38.000+01:00
diff --git a/TernaryBERT/gather_results.py b/TernaryBERT/gather_results.py
@@ -93,6 +93,7 @@ def gather_results(model_dir: str, task_name: str) -> Dict[str, Any]:
 
     data['parameters'] = parameters_num
     data['name'] = os.path.basename(model_dir)
+    data['model_name'] = 'TernaryBERT'
     print(data)
 
     return data
