pytorch坑点排雷

首页 > 技术文章 > pytorch坑点排雷

ichn 2018-01-24 13:35 原文

报GPU内存不够, 考虑改小batch_size
GPUTensor和CPUTensor不匹配, 查load模型的时候是否忘记layer = layer.cuda()

1
2

推荐阅读