【问题标题】:AI Text Generator returning AttributeError: can't set attributeAI 文本生成器返回 AttributeError:无法设置属性
【发布时间】:2023-02-08 02:50:13
【问题描述】:

我在 google colab 中使用https://github.com/minimaxir/aitextgen 的库 aitextgen 来制作文本生成器模型,但是当我运行“ai.train”时,它变成了一个错误。有什么建议可以解决这个问题吗?感谢您的帮助

这是代码

ai.train(file_name,
     line_by_line = False,
     from_cache = False,
     num_steps = 3000,
     generate_every = 1000,
     save_every = 1000,
     save_gdrive = False,
     learning_rate = 1e-3,
     fp16 = True, 
     batch_size = 1,
     )

这是错误信息

AttributeError                            Traceback (most recent call last)
<ipython-input-29-e611ad4fbd0c> in <module>
----> 1 ai.train(file_name,
      2          line_by_line = False,
      3          from_cache = False,
      4          num_steps = 3000,
      5          generate_every = 1000,

2 frames
/usr/local/lib/python3.8/dist-packages/aitextgen/aitextgen.py in train(self, train_data, output_dir, fp16, fp16_opt_level, n_gpu, tpu_cores, max_grad_norm, gradient_accumulation_steps, seed, learning_rate, weight_decay, adam_epsilon, warmup_steps, num_steps, save_every, generate_every, n_generate, loggers, batch_size, num_workers, benchmark, avg_loss_smoothing, save_gdrive, run_id, progress_bar_refresh_rate, freeze_layers, num_layers_freeze, use_deepspeed, **kwargs)
    673 
    674         # Wrap the model in a pytorch-lightning module
--> 675         train_model = ATGTransformer(self.model, train_data, hparams, self.tokenizer)
    676 
    677         # Begin training

/usr/local/lib/python3.8/dist-packages/aitextgen/train.py in __init__(self, model, dataset, hparams, tokenizer)
     19     def __init__(self, model, dataset, hparams, tokenizer):
     20         super(ATGTransformer, self).__init__()
---> 21         self.model, self.dataset, self.hparams, self.tokenizer = (
     22             model,
     23             dataset,

/usr/local/lib/python3.8/dist-packages/torch/nn/modules/module.py in __setattr__(self, name, value)
   1311                     buffers[name] = value
   1312                 else:
-> 1313                     super().__setattr__(name, value)
   1314 
   1315     def __delattr__(self, name):

AttributeError: can't set attribute

我已经尝试更改 torch、pytorch-lightning 和 aitextgen 的版本,但仍然出错。

【问题讨论】:

  • 嗨 Ahmad,提供生成错误的代码的 sn-p 将非常有用,因为还知道 python 的版本和涉及的库。 2023 年快乐
  • 请提供足够的代码,以便其他人可以更好地理解或重现问题。
  • 感谢您的建议,我已经添加了所用库的代码和链接。我在 google colab 中使用 python 3

标签: python-3.x machine-learning nlp artificial-intelligence google-colaboratory


【解决方案1】:

我有同样的错误,我所做的是: !pip 安装 pytorch-lightning==1.2.10 它再次起作用

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2015-02-08
    • 2018-08-30
    • 2014-08-09
    • 2017-07-17
    • 2014-04-06
    • 1970-01-01
    • 1970-01-01
    • 2021-06-14
    相关资源
    最近更新 更多