briscoooe commited on
Commit
a865492
·
verified ·
1 Parent(s): b1b216e

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. code/train_value_iteration.py +13 -5
code/train_value_iteration.py CHANGED
@@ -186,12 +186,20 @@ def main():
186
  help="States are sampled from walks of 1..this many moves. Past "
187
  "~20 every state is fully mixed (God's Number), so a larger "
188
  "value buys nothing.")
189
- p.add_argument("--update-threshold", type=float, default=0.05,
190
  help="Refresh the target network once training loss falls below "
191
- "this. Lower is more stable and slower.")
192
- p.add_argument("--min-update-steps", type=int, default=500,
193
- help="Never refresh the target network more often than this, so "
194
- "an early lucky batch cannot ratchet it forward.")
 
 
 
 
 
 
 
 
195
  p.add_argument("--eval-every", type=int, default=2000)
196
  p.add_argument("--eval-n", type=int, default=200)
197
  p.add_argument("--eval-depths", default="3,6,10,15,20")
 
186
  help="States are sampled from walks of 1..this many moves. Past "
187
  "~20 every state is fully mixed (God's Number), so a larger "
188
  "value buys nothing.")
189
+ p.add_argument("--update-threshold", type=float, default=0.01,
190
  help="Refresh the target network once training loss falls below "
191
+ "this. This is meant to be the binding constraint: it lets "
192
+ "an easy early target move on fast and gives a harder later "
193
+ "one as long as it needs. Set it above the loss the net "
194
+ "actually reaches and it never binds, leaving --min-update-"
195
+ "steps to pin the cadence -- measured at 0.05, the net sat "
196
+ "at 0.0004 and idled ~800 of every 1000 steps.")
197
+ p.add_argument("--min-update-steps", type=int, default=200,
198
+ help="Floor on the refresh interval, so one lucky batch cannot "
199
+ "ratchet the target forward. Each refresh propagates "
200
+ "accuracy roughly one move further from solved, so this "
201
+ "floor times ~30 is the run length; measured re-convergence "
202
+ "after a refresh is ~200 steps.")
203
  p.add_argument("--eval-every", type=int, default=2000)
204
  p.add_argument("--eval-n", type=int, default=200)
205
  p.add_argument("--eval-depths", default="3,6,10,15,20")