一开始没有用GPU在跑,直接用了CPU,模型最后接了两个2048的全连接层,运行直接报错。
解决办法:
1、减少全连接层参数。
x = Dense(512,activation='relu')(x)
2、batch改小一点。
batch_size=10
3、用GPU跑代码训练。
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"#设置根据电脑对应的接口ID选择设备
os.environ["CUDA_VISIBLE_DEVICES"]="0,1"#调用编号为0和1的gpu