DDPM玩具代码调试过程(一)

最新推荐文章于 2024-09-17 21:20:59 发布

清风拂山岗1111

最新推荐文章于 2024-09-17 21:20:59 发布

阅读量833

点赞数 1

CC 4.0 BY-SA版权

文章标签：笔记

原文链接：https://blog.youkuaiyun.com/weixin_43164054/article/details/124156206；https://blog.youkuaiyun.com/weixin_43835996/article/details/126955917；https://blog.youkuaiyun.com/weixin_39450145/article/details/104901193

本文介绍了在使用DDPM代码时遇到的四个常见问题：1.import错误关于相对引用的修正；2.wandb初始化时ValueError和project_name未设置；3.wandb实体参数引发的错误；4.multiprocessing导致的AttributeError。通过修改配置和代码片段，解决了这些问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

DDPM玩具代码来源：https://github.com/abarankab/DDPM

1.import报错：

不能用相对引用，需要把from .ema impot xxx 改成 from ema import xxx。所有类似的相对引用都需要修改。

2.报错：ValueError: args.log_to_wandb set to True but args.project_name is None

该报错信息是因为wandb没有注册的原因，参考链接wandb使用方法以及具体设置_wandb.init_神晟的光辉的博客-优快云博客https://blog.youkuaiyun.com/weixin_43164054/article/details/124156206

注册wandb账号并在当前虚拟环境下输入wandb login，将key复制后输入即可正常运行

3.wandb初始化报错

将初始化步骤中的entity参数删除就可以了，参考使用wandb报错：ERROR Error while calling W&B API: project not found (＜Response [404]＞)_error error while calling w&b api: user is not log_菜包好次的博客-优快云博客https://blog.youkuaiyun.com/weixin_43835996/article/details/126955917

wandb.init(entity='',
           project='',
           name=args.exp_name,
           config=args)
改为：

wandb.init(project='',
name=args.exp_name,
config=args)

4.报错：AttributeError: Can't pickle local object 'BaseDataset.img_transformer.<locals>.<lambda>'.............
EOFError: Ran out of input

搜索项目里的所有.py代码文件，一个文件的Ctrl+F搜索，看代码里面是否含有workers、mutilprocessing这两个关键字。搜索到workers将其值改为0即可，参考AttributeError: Can't pickle local object 'BaseDataset.img_transformer..'_weixin_39450145的博客-优快云博客https://blog.youkuaiyun.com/weixin_39450145/article/details/104901193

train_loader = script_utils.cycle(DataLoader(
    train_dataset,
    batch_size=batch_size,
    shuffle=True,
    drop_last=True,
    num_workers=2,
))

将其中的num_workers=0，问题解决