SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Changed string so that Circle CI accepts the warning

Browse Source
This commit is contained in:
LysandreJik
2019-08-21 21:24:48 -04:00
parent 87747518e9
commit 5c2b94c82a
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pytorch_transformers/modeling_utils.py
View File

@@ -381,8 +381,9 @@ class PreTrainedModel(nn.Module):
self.config.pruned_heads[int(layer)].append(head) self.config.pruned_heads[int(layer)].append(head)
to_be_pruned[int(layer)].append(head) to_be_pruned[int(layer)].append(head)
else: else:
logger.warning(f"Tried to remove head {head} of layer {layer} but it was already removed. " logger.warning("Tried to remove head " + head +
f"The removed heads are {heads_to_prune}") " of layer " + layer +
" but it was already removed. The current removed heads are " + heads_to_prune)
base_model._prune_heads(to_be_pruned) base_model._prune_heads(to_be_pruned)