Update hongik/engine_ai.py
Browse files- hongik/engine_ai.py +1 -1
hongik/engine_ai.py
CHANGED
|
@@ -46,7 +46,7 @@ class HongikAIEngine(BaseEngine):
|
|
| 46 |
It orchestrates the entire reinforcement learning loop.
|
| 47 |
"""
|
| 48 |
BOARD_SIZE, NUM_LAYERS, D_MODEL, NUM_HEADS, D_FF = 19, 7, 256, 8, 1024
|
| 49 |
-
SAVE_WEIGHTS_EVERY_STEPS, EVALUATION_EVERY_STEPS = 5,
|
| 50 |
REPLAY_BUFFER_SIZE, TRAINING_BATCH_SIZE = 200000, 32
|
| 51 |
CHECKPOINT_EVERY_GAMES = 10
|
| 52 |
|
|
|
|
| 46 |
It orchestrates the entire reinforcement learning loop.
|
| 47 |
"""
|
| 48 |
BOARD_SIZE, NUM_LAYERS, D_MODEL, NUM_HEADS, D_FF = 19, 7, 256, 8, 1024
|
| 49 |
+
SAVE_WEIGHTS_EVERY_STEPS, EVALUATION_EVERY_STEPS = 5, 50
|
| 50 |
REPLAY_BUFFER_SIZE, TRAINING_BATCH_SIZE = 200000, 32
|
| 51 |
CHECKPOINT_EVERY_GAMES = 10
|
| 52 |
|