Skip to content

Instantly share code, notes, and snippets.

@AMZzee

AMZzee/train_shape.py Secret

Created June 22, 2021 16:20
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save AMZzee/1d3642075735389570a8ad3fc86ecad1 to your computer and use it in GitHub Desktop.
Save AMZzee/1d3642075735389570a8ad3fc86ecad1 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
train_shape.py
finetune_model.summary()
finetune_model.compile(optimizer='Adam',loss='categorical_crossentropy',metrics=['accuracy'])
step_size_train=train_generator.n//train_generator.batch_size
history = finetune_model.fit_generator(generator=train_generator,steps_per_epoch=step_size_train,epochs=EPOCHS, shuffle=True)
finetune_model.save('shape_model.h5')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment