Skip to content

Commit 7a977d6

Browse files
committed
feat: capitilize model name
1 parent 8173a5c commit 7a977d6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

DynaBERT/gather_results.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ def gather_results(model_dir: str, task_name: str, depth_mult: float, width_mult
103103

104104
data['parameters'] = parameters_num
105105
data['name'] = os.path.basename(model_dir)
106-
data['model_name'] = 'dynabert_d_' + str(depth_mult) + '_w_' + str(width_mult)
106+
data['model_name'] = 'Dynabert_d_' + str(depth_mult) + '_w_' + str(width_mult)
107107
print(data)
108108
return data
109109

0 commit comments

Comments
 (0)