minivision-ai / photo2cartoon

人像卡通化探索项目 (photo-to-cartoon translation project)
MIT License
3.95k stars 764 forks source link

无响应 #65

Closed fanyushuai closed 2 years ago

fanyushuai commented 2 years ago

你好,一开始我找了100多张图,然后开始训练,内存溢出,然后我改为1张图,但是打印training start ! 之后就没反应了,请问是怎么回事?

hao-qiang commented 2 years ago

如果是用GPU训练,显存占用量和batch size有关,和训练数据量无关。可能是显存不足导致的。

fanyushuai commented 2 years ago

谢谢