CrowdCounting-P2PNet
CrowdCounting-P2PNet copied to clipboard
训练过程中,显存忽高忽低
训练过程中,显存会随着epoch忽高忽低。有时会导致显卡爆炸。 有人遇到这种问题吗?
您好~我也遇到相同问题,请问您现在解决了吗?
我也是,有谁知道如何解决码?
可以把匹配过程放到cpu上做
匹配过程重写就行,按batch分开写,不要写成一坨,浪费时间又浪费空间。