upweighting -- will exploit more & explore somewhat less

app.py

@@ -144,7 +144,7 @@ def next_image(embs, ys, calibrate_prompts):
             lin_class.coef_ = (lin_class.coef_.flatten() / (lin_class.coef_.flatten().norm())).unsqueeze(0)
 
             rng_prompt = random.choice(prompt_list)
-            w = 1# if len(embs) % 2 == 0 else 0
+            w = 1.4# if len(embs) % 2 == 0 else 0
             im_emb = w * lin_class.coef_.to(dtype=torch.float16)
             prompt= 'an image' if glob_idx % 2 == 0 else rng_prompt
             print(prompt, len(ys))