[BIG] pytorch-transformers => transformers
This commit is contained in:
@@ -15,7 +15,7 @@
|
||||
"""
|
||||
Preprocessing script before training DistilBERT.
|
||||
"""
|
||||
from pytorch_transformers import BertForPreTraining
|
||||
from transformers import BertForPreTraining
|
||||
import torch
|
||||
import argparse
|
||||
|
||||
|
||||
Reference in New Issue
Block a user