Base for loopback lora training setup, still working on proper sliders

2023-07-21 18:26:02 -06:00
parent aa13251877
commit ddcd9069e1
8 changed files with 1097 additions and 13 deletions
--- a/toolkit/losses.py
+++ b/toolkit/losses.py
@@ -83,3 +83,5 @@ class PatternLoss(torch.nn.Module):
        g_chan_loss = torch.abs(separated_chan_loss(g_chans) - separated_chan_loss(g_chans_target))
        b_chan_loss = torch.abs(separated_chan_loss(b_chans) - separated_chan_loss(b_chans_target))
        return (r_chan_loss + g_chan_loss + b_chan_loss) * 0.3333
+
+