pytorch坑点排雷

  • 报GPU内存不够, 考虑改小batch_size
  • GPUTensor和CPUTensor不匹配, 查load模型的时候是否忘记layer = layer.cuda()
posted @ 2018-01-24 13:35  ichneumon  阅读(330)  评论(0)  编辑  收藏  举报