feat: uhm, i changed some things

2025-11-25 20:20:08 +01:00 · 2025-11-25 20:20:08 +01:00 · 6de4db24cc
commit 6de4db24cc
parent b58682cb49
27 changed files with 1302 additions and 137 deletions
--- a/integer_discrete_flows/optimization/training.py
+++ b/integer_discrete_flows/optimization/training.py
@ -17,7 +17,7 @@ def train(epoch, train_loader, model, opt, args):
    for batch_idx, (data, _) in enumerate(train_loader):
        data = data.view(-1, *args.input_size)

-        data = data.to(args.device)
+        data = data.to(args.DEVICE)

        opt.zero_grad()
        loss, bpd, bpd_per_prior, pz, z, pys, py, ldj = model(data)