时间戳

首页 > 解决方案 > 模型甚至没有开始训练谷歌 Colab

python - 模型甚至没有开始训练谷歌 Colab

问题描述

我购买了 Colab pro 来训练我的 CNN 模型，当我尝试仅使用 4k 图像训练模型时，训练会立即开始......但是当我尝试使用 30k 图像训练模型时，训练甚至不会开始。.我什至等了 1 个小时，但模型从未开始训练它在第一个时期本身就卡住了……我的代码没有任何问题，我仔细检查了它……图片显示了我什至卡住的地方在我的模型训练的第一个时期，使用 30k 图像

标签： pythontensorflowkerasconv-neural-networkgoogle-colaboratory

解决方案

我使用这个生成器修复了那个大数据集问题...下面我用它的代码

train_datagen = ImageDataGenerator(rescale=1./255,
shear_range=0.2,
zoom_range=0.2,
horizontal_flip=True,
validation_split=0.2) # set validation split

train_generator = train_datagen.flow_from_directory(
data_dir,
target_size=(img_height, img_width),
batch_size=batch_size,
class_mode='categorical',
subset='training') # set as training data

validation_generator = train_datagen.flow_from_directory(
data_dir, # same directory as training data
target_size=(img_height, img_width),
batch_size=batch_size,
class_mode='categorical',
subset='validation') # set as validation data

history=model.fit_generator(
     train_generator,
     steps_per_epoch = train_generator.samples // batch_size,
     validation_data = validation_generator, 
     validation_steps = validation_generator.samples // batch_size,
     epochs = epochs)

推荐阅读