Skip to content Navigation Menu Toggle navigation ...
2、数据集问题,小于500的自行考虑增加数据集,同时测试不同的模型 ... b、显存不足问题 问:为什么我运行train.py下面的命令行闪的贼快,还提示OOM啥的? 答:这是在keras中出现的,爆显存了,可以改小batch_size。 需要注意的是,受到BatchNorm2d影响,batch_size ...