dorimedini / robustness_properties

GLOBIGLOBIGLOBI
0 stars 0 forks source link

added CNNTrainer with a simple CNN for cifar10 (thanks noam kapara) #116

Closed galshachaf closed 5 years ago

galshachaf commented 5 years ago

Signed-off-by: galshachaf galshac@gmail.com

galshachaf commented 5 years ago

Mm good question, don't know

On Tue, Sep 3, 2019, 11:26 dorimedini notifications@github.com wrote:

@dorimedini approved this pull request.

In trainer.py https://github.com/dorimedini/kardashigans/pull/116#discussion_r320147745 :

  • model.add(Conv2D(64, (3, 3)))
  • model.add(Activation('relu'))
  • model.add(MaxPooling2D(pool_size=(2, 2)))
  • model.add(Dropout(0.25))
  • model.add(Flatten())
  • model.add(Dense(512))
  • model.add(Activation('relu'))
  • model.add(Dropout(0.5))
  • model.add(Dense(self._n_classes))
  • model.add(Activation('softmax'))
  • return model
  • @classmethod
  • def get_weighted_layers_indices(cls):

In BaseTrainer this method isn't a class method, Does this "override" work as expected?

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub https://github.com/dorimedini/kardashigans/pull/116?email_source=notifications&email_token=ADHINCCYTQSPITN6WO33EJ3QHYNRZA5CNFSM4ITCQITKYY3PNVWWK3TUL52HS4DFWFIHK3DMKJSXC5LFON2FEZLWNFSXPKTDN5WW2ZLOORPWSZGOCDN3LXA#pullrequestreview-282834396, or mute the thread https://github.com/notifications/unsubscribe-auth/ADHINCDIUR7WWHGXHIP7E6TQHYNRZANCNFSM4ITCQITA .