dot-weight_style_layer_loss: Calculate the Weighted Style Layer Loss

Description Usage Arguments Value

Description

A simple convenience function to multiple loss by weight.

Usage

1
.weight_style_layer_loss(style_layer, generated_layer, lambda)

Arguments

style_layer

Features extracted from a single style layer.

generated_layer

Features extracted from a single generated layer. Should be the same layer as style_layer.

lambda

The weight to apply to this loss.

Value

A scalar torch_tensor representing the weighted loss for this layer.


jonthegeek/stylex documentation built on Dec. 21, 2021, 2:17 a.m.