Entrainement

code/resnet18/resnet18_snake.py

@@ -93,7 +93,7 @@ model.compile(sgd, loss='categorical_crossentropy', metrics=['accuracy'])
 history = model.fit(train_images,
          train_labels,
          batch_size=64,
-         epochs=4,
+         epochs=25,
          validation_data=(val_images, val_labels),
          )