Skip to content

Commit 9a56348

Browse files
committed
update patience and training rate
1 parent e68d98c commit 9a56348

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

odir_xception_training_basic.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@
3434
batch_size = 32
3535
num_classes = 8
3636
epochs = 100
37-
patience = 5
37+
patience = 3
3838

3939
# class_weight = {0: 1.,
4040
# 1: 1.583802025,
@@ -87,7 +87,7 @@
8787
# optimizer='rmsprop',
8888
# metrics=defined_metrics)
8989

90-
sgd = SGD(lr=0.01, decay=1e-6, momentum=0.9, nesterov=True)
90+
sgd = SGD(lr=0.001, decay=1e-6, momentum=0.9, nesterov=True)
9191
print('Configuration Start -------------------------')
9292
print(sgd.get_config())
9393
print('Configuration End -------------------------')

0 commit comments

Comments
 (0)