机器学习碰到MemoryError: Unable to allocate1.5 . MiB for an array with shape解决办法
详细代码如下# 训练集train_gen = train_val_generator( data_dir='../train', target_size=(64, 64), batch_size=32, class_model='categorical', subset='training')# 验证集val_gen = train_val_generator( data_dir='../test', target_size=(64, 64
原创
2021-03-16 15:15:48 ·
25788 阅读 ·
12 评论