解决方法： 

（1）之前设的gpu为1，batch size为1    后边改为gpu为2，batch size为8

 

 

   

 

 结果为