Fix: torch.utils.checkpoint import error. (#9626)

2021-01-18 12:33:39 +03:00
parent 72fc9abf17
commit 65eb5d9ac5
13 changed files with 13 additions and 0 deletions
--- a/src/transformers/models/roberta/modeling_roberta.py
+++ b/src/transformers/models/roberta/modeling_roberta.py
@@ -19,6 +19,7 @@ import math

 import torch
 import torch.nn as nn
+import torch.utils.checkpoint
 from torch.nn import CrossEntropyLoss, MSELoss

 from ...activations import ACT2FN, gelu