Skip to content

Commit

Permalink
perf: more stronger 🚧
Browse files Browse the repository at this point in the history
  • Loading branch information
Joker2770 committed Aug 17, 2024
1 parent b351d05 commit a7cf03c
Showing 1 changed file with 29 additions and 29 deletions.
58 changes: 29 additions & 29 deletions src/common.h
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
MIT License
Copyright (c) 2022 Augustusmyc
Copyright (c) 2023 Joker2770
Copyright (c) 2023-2024 Joker2770
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
Expand Down Expand Up @@ -32,45 +32,45 @@ SOFTWARE.
// #define USE_TENSORRT
// #define USE_ROCM

#define CHANNEL_SIZE 3
#define CHANNEL_SIZE (3)

#ifdef SMALL_BOARD_MODE
#define BOARD_SIZE 3
#define N_IN_ROW 3
#define NUM_MCT_THREADS 4
#define NUM_MCT_SIMS 54
#define EXPLORE_STEP 3
#define C_PUCT 3
#define C_VIRTUAL_LOSS 1
#define NUM_CHANNELS 64
#define NUM_LAYERS 2

#define BATCH_SIZE 64 // 512
#define DIRI 0.1

#define NUM_TRAIN_THREADS 70
#define BOARD_SIZE (3)
#define N_IN_ROW (3)
#define NUM_MCT_THREADS (4)
#define NUM_MCT_SIMS (54)
#define EXPLORE_STEP (3)
#define C_PUCT (3)
#define C_VIRTUAL_LOSS (1)
#define NUM_CHANNELS (64)
#define NUM_LAYERS (2)

#define BATCH_SIZE (64) // 512
#define DIRI (0.1)

#define NUM_TRAIN_THREADS (70)
#else
#define BOARD_SIZE 15
#define N_IN_ROW 5
#define NUM_MCT_THREADS 4
#define NUM_MCT_SIMS 1600
#define BOARD_SIZE (15)
#define N_IN_ROW (5)
#define NUM_MCT_THREADS (4)
#define NUM_MCT_SIMS (1600)
#define EXPLORE_STEP (BOARD_SIZE * BOARD_SIZE)
#define C_PUCT 5
#define C_VIRTUAL_LOSS 3
// #define NUM_CHANNELS 256
// #define NUM_LAYERS 4
#define C_PUCT (5)
#define C_VIRTUAL_LOSS (3)
// #define NUM_CHANNELS (256)
// #define NUM_LAYERS (4)

#define BATCH_SIZE 256
#define DIRI 0.01
#define BATCH_SIZE (256)
#define DIRI (0.01)

#define NUM_TRAIN_THREADS 10
#define NUM_TRAIN_THREADS (10)

// 0 - free-style, 1 - standard, 2 - continuous, 4 - renju, 8 - caro
#define DEFAULT_RULE 0
#define DEFAULT_RULE (0)

#endif

#define BLACK 1
#define BLACK (1)
#define WHITE (-BLACK)

#define BUFFER_LEN (BOARD_SIZE * BOARD_SIZE + 1)
Expand Down

0 comments on commit a7cf03c

Please sign in to comment.