history = VGG_model.fit(train_dataset,epochs=config["EPOCHS"],validation_data=valid_dataset,callbacks=[early_stopping_cb,checkpoint_cb,
                                          WandbMetricsLogger(log_freq=10)])

        __
        __