查阅资料多处显示是因为是yarn的配置问题。内存与CPU的资源不够，更改更改yarn中的设置，添加以下内容；在分发到集群，解决问题。|yarn.nodemanager.resource.memory-mb20480yarn.scheduler.minimum-allocation-mb2048yarn.nodemanager.vmem-pmem-ratio2.1...

出现ImportError: cannot import name ‘_obtain_input_shape’ from ‘keras.applications.imagenet_utils’_obtain_input_shape换地方了。

这个报错是说Torch找不到cuda ，无法编译，要调用CPU环境。然后使用快捷键ctrl+R将.cuda()改成.to(device)