feat: capitilize model name

huawei-noah · Dec 17, 2021 · 7a977d6 · 7a977d6
1 parent 8173a5c
commit 7a977d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/DynaBERT/gather_results.py b/DynaBERT/gather_results.py
@@ -103,7 +103,7 @@ def gather_results(model_dir: str, task_name: str, depth_mult: float, width_mult
 
     data['parameters'] = parameters_num
     data['name'] = os.path.basename(model_dir)
-    data['model_name'] = 'dynabert_d_' + str(depth_mult) + '_w_' + str(width_mult)
+    data['model_name'] = 'Dynabert_d_' + str(depth_mult) + '_w_' + str(width_mult)
     print(data)
     return data