diff --git a/modules/sampling/attention_bandit/__init__.py b/modules/sampling/attention_bandit/__init__.py index 1866bbe4..f86cfd97 100644 --- a/modules/sampling/attention_bandit/__init__.py +++ b/modules/sampling/attention_bandit/__init__.py @@ -66,7 +66,7 @@ GROUPS = 4 ARMS = (-1.0, 0.0, 1.0) STAY = 1 # index of "as is" STAY_HEAD_START = 2.0 # extra likes "as is" starts with, so a new key is stable -MAX_ROWS = 200 +MAX_ROWS = 10 ** 9 # no cap: a rated run is under 1 KB START_SPREAD = 0.2 SPREAD_MIN, SPREAD_MAX = 0.05, 0.5 LR = 0.5