Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -56,10 +56,11 @@ print(f"[SYSTEM 24.4] Compute: {DEVICE.type.upper()} | RL Bandit + Native VM")
|
|
| 56 |
# ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 57 |
# SECTION 1: CONSTANTS
|
| 58 |
# ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 59 |
-
SOLVE_THRESHOLD = 0.05
|
| 60 |
-
VERIFY_G1_THRESHOLD = 0.08
|
| 61 |
VERIFY_G2_TOLERANCE = 1e-3
|
| 62 |
-
|
|
|
|
| 63 |
N_MPRT_EXPLORE = 1000
|
| 64 |
PROJECTION_CACHE: Dict = {}
|
| 65 |
|
|
@@ -694,7 +695,7 @@ def _batched_deduce_and_evaluate(problem: Problem, hyps: List['Hypothesis']) ->
|
|
| 694 |
X = torch.tensor(x_data, device=DEVICE, dtype=torch.float32, requires_grad=True)
|
| 695 |
mask = torch.tensor(mask_data, device=DEVICE, dtype=torch.float32)
|
| 696 |
target = torch.tensor(target_data, device=DEVICE, dtype=torch.float32)
|
| 697 |
-
optimizer = torch.optim.Adam([X], lr=0.
|
| 698 |
|
| 699 |
for step in range(DEDUCE_ADAM_STEPS):
|
| 700 |
optimizer.zero_grad()
|
|
|
|
| 56 |
# ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 57 |
# SECTION 1: CONSTANTS
|
| 58 |
# ββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 59 |
+
SOLVE_THRESHOLD = 0.001 # Was 0.05
|
| 60 |
+
VERIFY_G1_THRESHOLD = 0.005 # Was 0.08
|
| 61 |
VERIFY_G2_TOLERANCE = 1e-3
|
| 62 |
+
|
| 63 |
+
DEDUCE_ADAM_STEPS = 60
|
| 64 |
N_MPRT_EXPLORE = 1000
|
| 65 |
PROJECTION_CACHE: Dict = {}
|
| 66 |
|
|
|
|
| 695 |
X = torch.tensor(x_data, device=DEVICE, dtype=torch.float32, requires_grad=True)
|
| 696 |
mask = torch.tensor(mask_data, device=DEVICE, dtype=torch.float32)
|
| 697 |
target = torch.tensor(target_data, device=DEVICE, dtype=torch.float32)
|
| 698 |
+
optimizer = torch.optim.Adam([X], lr=0.01) #0.05
|
| 699 |
|
| 700 |
for step in range(DEDUCE_ADAM_STEPS):
|
| 701 |
optimizer.zero_grad()
|