pytorch
文章平均质量分 50
wzc-run
不学习就会被AI替代
展开
-
遇到ModuleNotFoundError: No module named xxx
会发现,路径只有conda的lib库以及当前运行文件的路径,也就是"xxx/code/scripts"train.py需要import src.utils as utils。此时,python解释器当然是搜索不到src文件夹下的文件了。显然,不是因为缺少包导致的问题。打印下当前文件的搜索路径。原创 2024-03-19 14:42:45 · 246 阅读 · 0 评论 -
HuggingFace Accelerate整合wandb记录实验
看了半天HuggingFace教程没看明白怎么添加其他wandb run的参数(我还是太菜了!),最后在wandb的教程中找到了…传入init_kwargs参数即可。原创 2023-03-24 01:24:05 · 1077 阅读 · 0 评论 -
HuggingFace Accelerate解决分布式训练
由于项目代码比较复杂且可读性差…,尝试使用Hugging Face的Accelerate实现多卡的分布式训练。原创 2023-03-24 01:11:19 · 5487 阅读 · 0 评论