Вы можете сделать это с помощью PhantomJSDriver.
PhantomJSDriver pd = ((PhantomJSDriver) ((WebDriverFacade) getDriver()).getProxiedDriver());
pd.executePhantomJS(
"this.onResourceRequested = function(request, net) {" +
" net.setHeader('header-name', 'header-value')" +
"};");
Используя объект запроса, вы также можете фильтровать так, чтобы заголовок не был установлен для каждого запроса.
Это означает, что все операции внутри вашей пользовательской функции потерь должны быть дифференцируемыми, поскольку в противном случае процедура оптимизации не может быть выполнена. Для этого вам просто нужно один за другим проверить, какая операция является преступником в вашем коде, и заменить ее дифференцируемым внутренним аналогом Keras или найти какую-то другую альтернативу.
Учитывая предоставленный фрагмент кода, может быть несколько возможных предложений, чтобы он работал:
Надеюсь, это поможет