jingyaogong
Results
2
comments of
jingyaogong
> > 可以修改一下这个函数, utils.py里面： > > ```python > > def auto_configure_device_map(num_gpus: int) -> Dict[str, int]: > > # transformer.word_embeddings 占用1层 > > # transformer.final_layernorm 和 lm_head 占用1层 > > #...
1.清华大学云盘 [https://cloud.tsinghua.edu.cn/d/674208019e314311ab5c/](url) 2.百度网盘 [https://pan.baidu.com/s/1rn1ZtjgfMZg_Kw4PNd3jVQ?pwd=aiha](url)