ludwig AttributeError: module 'torchtext' has no attribute 'transforms'

liwlm1x9  于 2个月前  发布在  其他
关注(0)|答案(1)|浏览(40)

描述bug

在尝试使用encoder = bert训练模型时,我遇到了这个错误。

重现问题

trainer:
    validation_field: engagement_label
    validation_metric: roc_auc
    early_stop: 10

input_features:
    - name: text1
      type: text
      encoder: 
          type: bert
    - name: text2
      type: text
      encoder: 
          type: bert

output_features:
    - name: engagement_label
      type: binary
      threshold: 0.6

预期行为

模型应该可以正常训练,不会出现错误。

截图

环境信息(请填写以下信息):

  • OS: [例如iOS]
  • 版本[例如22]
  • Python 3.11.8
  • ludwig==0.10.3
    其他上下文信息

在这里添加有关问题的任何其他上下文信息。

相关问题