Возврат потерь при передаче в тензорном стиле 0

    x_content = model['conv4_2']
    x_style = [model['conv' + str(i) + '_1'] for i in range(1, 6)]

    sess.run(model['input'].assign(content_image))
    y_content = model['conv4_2']
    content_loss = loss.content_loss(x_content, y_content)

    sess.run(model['input'].assign(style_image))
    y_style = [model['conv' + str(i) + '_1'] for i in range(1, 6)]
    style_loss = loss.style_loss(x_style, y_style)

    total_loss = args.ALPHA * content_loss + args.BETA * style_loss

Я изучаю машинное обучение о нейронном алгоритме художественного образа. Но он не очень хорошо оптимизировался, также content_loss и потеря стиля вернули 0.

Почему это возвращается 0. Вот моя ссылка на git-репо https://github.com/zhsir123/Artistic-Image

Источник

0 ответов

Другие вопросы по тегам