gram[source]

gram(x)

content_loss[source]

content_loss(pred, targ, fts, layer_ws=None)

style_loss[source]

style_loss(pred, targ, fts, layer_ws=None, stl_ws=None)

tv_loss[source]

tv_loss(pred, *_)

pixel_loss[source]

pixel_loss(pred, targ, fts, base_loss='mse_loss', **kwargs)

class FastStyleLoss[source]

FastStyleLoss(stl_w=300000.0, cnt_w=1, tv_w=20, stl_loss=None, cnt_loss=None, tv_loss=None)

class SuperResLoss[source]

SuperResLoss(pixel_loss=None, cnt_loss=None)