Adding proper attribution

2026-08-29 01:55:46 +08:00 · 2020-06-18 19:17:11 -07:00
parent c6f28b9557
commit ee1f8d2f3b
1 changed files with 1 additions and 1 deletions
@@ -43,7 +43,7 @@ class FeatureLoss(nn.Module):
        self.hooks.remove()


-# Includes wasserstein loss
+# Refactored code, originally from https://github.com/VinceMarron/style_transfer
 class WassFeatureLoss(nn.Module):
    def __init__(self, layer_wgts=[5, 15, 2], wass_wgts=[3.0, 0.7, 0.01]):
        super().__init__()