da4848e1156edd2076b881ce2300ae2d3cc18246,examples/train_pascal.py,,,#,51

Before Change

    // start training
    model.fit_generator(
        generator=train_generator,
        steps_per_epoch=len(train_generator.image_names) // args.batch_size,
        epochs=50,
        verbose=1,
        validation_data=val_generator,

After Change

    )

    // start training
    training_model.fit_generator(
        generator=train_generator,
        steps_per_epoch=10000,
        epochs=50,
        verbose=1,
        validation_data=val_generator,
        validation_steps=3000,  // len(val_generator.image_names) // args.batch_size,
        callbacks=create_callbacks(model, training_model, prediction_model),
    )

    // store final result too
    model.save(os.path.join("snapshots", "resnet50_voc_final.h5"))

Instances