错误解决方法
排查一:
把网络的各个shape打印出来,发现其实是数据shape不匹配。
有问题先去看数据、embedding、网络维度!!!
排查二:
如果shape都是对应的,但是还是报相同的错误,应该是GPU内存不够,可以换一个空闲的GPU。或者调小Batch都可以。
错误解决方法
排查一:
把网络的各个shape打印出来,发现其实是数据shape不匹配。
有问题先去看数据、embedding、网络维度!!!
排查二:
如果shape都是对应的,但是还是报相同的错误,应该是GPU内存不够,可以换一个空闲的GPU。或者调小Batch都可以。