From 1ad8c606557cd09815bd14f422c0b7d674874d17 Mon Sep 17 00:00:00 2001
From: Varuna Jayasiri Loop for Track data for analytics Save checkpoints every $1,000$ iterations Initialize CFR algorithmepochs
times683 for t in monit.loop(self.epochs):
683 for t in monit.iterate('Train', self.epochs):
689 self.tracker(self.info_sets)
-690 tracker.save()
689 tracker.add_global_step()
+690 self.tracker(self.info_sets)
+691 tracker.save()
693 if (t + 1) % 1_000 == 0:
-694 experiment.save_checkpoint()
-695 tracker.new_line()
694 if (t + 1) % 1_000 == 0:
+695 experiment.save_checkpoint()
719 with monit.section("Track"):
-720 for I in info_sets.values():
-721 avg_strategy = I.get_average_strategy()
-722 for a in I.actions():
-723 tracker.add({
-724 f'strategy.{I.key}.{a}': I.strategy[a],
-725 f'average_strategy.{I.key}.{a}': avg_strategy[a],
-726 f'regret.{I.key}.{a}': I.regret[a],
-727 })
719 for I in info_sets.values():
+720 avg_strategy = I.get_average_strategy()
+721 for a in I.actions():
+722 tracker.add({
+723 f'strategy.{I.key}.{a}': I.strategy[a],
+724 f'average_strategy.{I.key}.{a}': avg_strategy[a],
+725 f'regret.{I.key}.{a}': I.regret[a],
+726 })
Configurable CFR module
730class CFRConfigs(BaseConfigs):
729class CFRConfigs(BaseConfigs):
734 create_new_history: Callable[[], History]
-735 epochs: int = 1_00_000
-736 cfr: CFR = 'simple_cfr'
733 create_new_history: Callable[[], History]
+734 epochs: int = 1_00_000
+735 cfr: CFR = 'simple_cfr'
739@option(CFRConfigs.cfr)
-740def simple_cfr(c: CFRConfigs):
738@option(CFRConfigs.cfr)
+739def simple_cfr(c: CFRConfigs):
744 return CFR(create_new_history=c.create_new_history,
-745 epochs=c.epochs)
743 return CFR(create_new_history=c.create_new_history,
+744 epochs=c.epochs)
235 experiment.create(name='kuhn_poker', writers={'sqlite', 'screen'})
235 experiment.create(name='kuhn_poker', writers={'sqlite'})