Hi There,
This is Santhosh, While I’m running the neural networks in laptop it is taking arround 46mins to run each epoch. So please me in suggesting best free gpu platform where i can run my code fast enough.
Regards,
Santhosh.
Load the base model
base_model = Xception(weights=‘imagenet’, include_top=False, input_shape=IMG_SIZE + (3,))
Add new layers on top of the base model
x = base_model.output
x = GlobalAveragePooling2D()(x)
x = Dense(1024, activation=‘relu’)(x)
x = BatchNormalization()(x)
x = Dropout(0.5)(x)
x = Dense(512, activation=‘relu’)(x)
x = BatchNormalization()(x)
x = Dropout(0.5)(x)
predictions = Dense(1, activation=‘sigmoid’)(x)
This is the model we will train
model = Model(inputs=base_model.input, outputs=predictions)
Freeze the layers of the base model
for layer in base_model.layers:
layer.trainable = False
Compile the model
model.compile(optimizer=Adam(), loss=‘binary_crossentropy’, metrics=[‘accuracy’])