tensorflow - 训练 distilbert-base-uncased 模型时出错

Question

我在处理 distilbert-base-uncased 分类器时遇到以下错误。

• 'NoneType' 和 'int' 的实例之间不支持 '>' 这里是代码 -

  from transformers import TFDistilBertForSequenceClassification, TFTrainer, 
  TFTrainingArguments
  
    training_args = TFTrainingArguments(
    output_dir='./results',          # output directory
    num_train_epochs=10,              # total number of training epochs
    per_device_train_batch_size=8,  # batch size per device during training
    per_device_eval_batch_size=16,   # batch size for evaluation
    warmup_steps=500,                # number of warmup steps for learning rate scheduler
    weight_decay=0.01,               # strength of weight decay
    logging_dir='./logs',            # directory for storing logs
    logging_steps=10,
  
    )
  
  with training_args.strategy.scope():
    model =TFDistilBertForSequenceClassification.from_pretrained("distilbert-base-uncased")
  
    trainer = TFTrainer(
    model=model,                         # the instantiated  Transformers model to be trained
    args=training_args,                  # training arguments, 
   defined above
    train_dataset=train_dataset,         # training dataset
    eval_dataset=valid_dataset             # evaluation dataset
    )
  
    trainer.train()