Update pytorch_weights_postprocess.py
Browse files
pytorch_weights_postprocess.py
CHANGED
|
@@ -55,8 +55,8 @@ for i in range(12):
|
|
| 55 |
|
| 56 |
#Pooler
|
| 57 |
for k, v in full_state_dict.items():
|
| 58 |
-
if "
|
| 59 |
-
mod_dict.update({k:v})
|
| 60 |
|
| 61 |
|
| 62 |
for k, v in mod_dict.items():
|
|
|
|
| 55 |
|
| 56 |
#Pooler
|
| 57 |
for k, v in full_state_dict.items():
|
| 58 |
+
if "dense" in k:
|
| 59 |
+
mod_dict.update({f"pooler.{k}":v})
|
| 60 |
|
| 61 |
|
| 62 |
for k, v in mod_dict.items():
|