history = model.fit(normalized_train_dataset,validation_data=normalized_valid_dataset,
                    epochs=config["epochs"],callbacks=[early_stopping_cb,checkpoint_cb,
                                          tensorboard_cb,WandbMetricsLogger(log_freq=10)])

        __
        __