বিবরণ:
জিপিইউ : জিটিএক্স 1080
প্রশিক্ষণ : 10 ক্লাসের অন্তর্ভুক্ত ~ 1.1 মিলিয়ন ডলার
বৈধতা : 10 শ্রেণীর অন্তর্ভুক্ত। 150 হাজার চিত্র images
প্রতি যুগের সময় : hours 10 ঘন্টা
আমি সিইউডিএ, সিইউডিএনএন এবং টেনসরফ্লো (টেনসরফ্লো জিপিইউ) সেটআপ করেছি।
আমি মনে করি না যে আমার মডেলটি এমন জটিল যে প্রতি যুগের জন্য 10 ঘন্টা সময় লাগে। এমনকি আমার জিপিইউ সমস্যা কিনা তা যাচাই করেছিলাম তবে তা হয়নি।
সম্পূর্ণ সংযুক্ত স্তরগুলির কারণে প্রশিক্ষণের সময় কি?
আমার মডেল:
model = Sequential()
model.add()
model.add(Conv2D(64, (3, 3), padding="same", strides=2))
model.add(Activation('relu'))
model.add(Dropout(0.25))
model.add(Conv2D(64, (3, 3), padding="same", strides=2))
model.add(Activation('relu'))
model.add(Dropout(0.25))
model.add(Conv2D(32, (3, 3)))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(3, 3), strides=2))
model.add(Flatten())
model.add(Dense(256))
model.add(Activation('relu'))
model.add(Dense(4096))
model.add(Activation('relu'))
model.add(Dense(10))
model.add(Activation('softmax'))
model.summary()
opt = keras.optimizers.rmsprop(lr=0.0001, decay=1e-6)
model.compile(loss='categorical_crossentropy',
optimizer=opt,
metrics=['accuracy']
)
কারণ অনেকগুলি ডেটা আমি ইমেজডেটা জেনারেটর ব্যবহার করেছি used
gen = ImageDataGenerator(
horizontal_flip=True
)
train_gen = gen.flow_from_directory(
'train/',
target_size=(512, 512),
batch_size=5,
class_mode="categorical"
)
valid_gen = gen.flow_from_directory(
'validation/',
target_size=(512, 512),
batch_size=5,
class_mode="categorical"
)