From bc67103752fbbc1c259dcc2e31b164b92bde517c Mon Sep 17 00:00:00 2001 From: Martin Eyben Date: Fri, 16 May 2025 11:15:27 +0200 Subject: [PATCH] fix: create result file --- go_ia/main.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/go_ia/main.py b/go_ia/main.py index fdab1ed..42e559a 100755 --- a/go_ia/main.py +++ b/go_ia/main.py @@ -236,10 +236,13 @@ def stones_to_board(black_stones, white_stones, black_plays): def position_predict(black_stones, white_stones, depth): - board = stones_to_board(black_stones, white_stones, depth % 2 == 0) + board = torch.from_numpy(np.array([ + stones_to_board(black_stones, white_stones, depth % 2 == 0) + ])).float().to(device) + mymodel.eval() with torch.no_grad(): - prediction = mymodel(board.unsqueeze(0)) + prediction = mymodel(board) return prediction @@ -310,6 +313,7 @@ if __name__ == "__main__": with gzip.open(args.R) as fz: data = json.loads(fz.read().decode("utf-8")) create_result_file(data) + exit(0) if args.t is not None and args.T is not None: trainset = torch.load(args.t)