Description Usage Arguments Value
A simple convenience function to multiple loss by weight.
1 | .weight_style_layer_loss(style_layer, generated_layer, lambda)
|
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. |
A scalar torch_tensor representing the weighted loss for this layer.
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.