We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 00d4e9e commit e6addf2Copy full SHA for e6addf2
vgg_bn.py
@@ -72,7 +72,7 @@ def finetune(self):
72
73
def compile(self):
74
self.model.compile(optimizer=Adam(lr=self.lr),
75
- loss='categorical_crossentropy', metrics=['val_loss', 'accuracy'])
+ loss='categorical_crossentropy', metrics=['accuracy'])
76
77
def fit(self, trn_path, val_path, nb_trn_samples, nb_val_samples, nb_epoch=1, callbacks=None, aug=False):
78
if aug:
0 commit comments