前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
Created by: yeyupiaoling
我使用的是docker进行训练,显存为11G,在训练过程中会出现不断吃显存,直到显存不足挂掉。设置小batch_size也没有用的。