【发布时间】: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