More diverse data to train
@@ -17,13 +17,13 @@ import torch.nn.functional as F
from torch.autograd import Variable
-MCTS_SEARCHES = 5
-MCTS_BATCH_SIZE = 16
+MCTS_SEARCHES = 10
+MCTS_BATCH_SIZE = 10
REPLAY_BUFFER = 10000
LEARNING_RATE = 1e-4
BATCH_SIZE = 128
TRAIN_ROUNDS = 10
-MIN_REPLAY_TO_TRAIN = 200
+MIN_REPLAY_TO_TRAIN = TRAIN_ROUNDS * BATCH_SIZE
BEST_SCORES_HIST = 30
BEST_NET_WIN_RATIO = 0.55
