search

yucornetto / MGMatting

This repository includes the official project of Mask Guided (MG) Matting, presented in our paper: Mask Guided Matting via Progressive Refinement Network
Other
331 stars 46 forks source link

训练不稳定的问题 #32

Closed caihaunqai closed 2 years ago

caihaunqai commented 2 years ago

您好，我使用四块V100进行训练，只用了28的batch_size，训练过程中在开始或其他时候显存会突然增大，增加的话会爆显存，所以无法增加。但是实际的占用显存只有不到1/3。您知道如何解决吗？谢谢！