错误如下：
RuntimeError: ResourceExhaustedError:
Out of memory error on GPU 0. Cannot allocate 2.679688GB memory on GPU 0, available memory is only 1.532593GB.
尝试调小batch-size也没有用
