各位大佬,想请教一下代码前面都正常,运行到df.to_csv就报错是哪出了问题啊

tutulovecat 2023-12-28 11:16:10

    mconf = GPTConfig(train_dataset.vocab_size, train_dataset.max_len, num_props=num_props,  # args.num_props,
                        n_layer=args.n_layer, n_head=args.n_head, n_embd=args.n_embd, scaffold=args.scaffold, scaffold_maxlen=scaffold_max_len,
                        lstm=args.lstm, lstm_layers=args.lstm_layers)
    model = GPT(mconf)

    tconf = TrainerConfig(max_epochs=args.max_epochs, batch_size=args.batch_size, learning_rate=args.learning_rate,
                            lr_decay=True, warmup_tokens=0.1*len(train_data)*max_len, final_tokens=args.max_epochs*len(train_data)*max_len,
                            num_workers=10, ckpt_path=f'D:/Jupyter/MolGPT/train-copy1/cond_gpt/weights/{args.run_name}.pt', block_size=train_dataset.max_len, generate=False)
    trainer = Trainer(model, train_dataset, valid_dataset,
                        tconf, train_dataset.stoi, train_dataset.itos)
    df = trainer.train(wandb)

    df.to_csv(f'D:/Jupyter/MolGPT/train-copy1/{args.run_name}.csv', index=False)

报错为

AttributeError                            Traceback (most recent call last)
Cell In[3], line 76
     72 trainer = Trainer(model, train_dataset, valid_dataset,
     73                     tconf, train_dataset.stoi, train_dataset.itos)
     74 df = trainer.train(wandb)
---> 76 df.to_csv(f'D:/Jupyter/MolGPT/train-copy1/{args.run_name}.csv', index=False)

AttributeError: 'NoneType' object has no attribute 'to_csv'
...全文
990 1 打赏 收藏 转发到动态 举报
写回复
用AI写文章
1 条回复
切换为时间正序
请发表友善的回复…
发表回复
T0BSky 2024-05-22
  • 打赏
  • 举报
回复

报错信息显示'NoneType' object has no attribute 'to_csv',意味着dfNone类型,而不是一个DataFrame对象,因此无法调用to_csv方法。这通常是因为trainer.train(wandb)返回了None,而不是一个DataFrame对象。你可以在trainer.train(wandb)之后添加一些调试语句,例如打印df,查看其返回值是什么,以便找出问题所在。可能需要检查Trainer.train方法的实现,确保它返回一个正确的DataFrame对象。

51,407

社区成员

发帖
与我相关
我的任务
社区描述
Java相关技术讨论
javaspring bootspring cloud 技术论坛(原bbs)
社区管理员
  • Java相关社区
  • 小虚竹
  • 谙忆
加入社区
  • 近7日
  • 近30日
  • 至今
社区公告
暂无公告

试试用AI创作助手写篇文章吧