katrain

katrain

Improve your Baduk skills by training with KataGo!

Stars: 1569

Visit
 screenshot

KaTrain is a tool designed for analyzing games and playing go with AI feedback from KataGo. Users can review their games to find costly moves, play against AI with immediate feedback, play against weakened AI versions, and generate focused SGF reviews. The tool provides various features such as previews, tutorials, installation instructions, and configuration options for KataGo. Users can play against AI, receive instant feedback on moves, explore variations, and request in-depth analysis. KaTrain also supports distributed training for contributing to KataGo's strength and training bigger models. The tool offers themes customization, FAQ section, and opportunities for support and contribution through GitHub issues and Discord community.

README:

KaTrain

Latest Release License:MIT GitHub Downloads PyPI Downloads Discord

KaTrain is a tool for analyzing games and playing go with AI feedback from KataGo:

  • Review your games to find the moves that were most costly in terms of points lost.
  • Play against AI and get immediate feedback on mistakes with option to retry.
  • Play against a wide range of weakened versions of AI with various styles.
  • Automatically generate focused SGF reviews which show your biggest mistakes.

Manual

English German French Russian Turkish

Simplified Chinese Traditional Chinese Korean Japanese

Preview and Youtube Videos

screenshot

Local Joseki Analysis Analysis Tutorial Teaching Game Tutorial
Local Joseki Analysis Video Analysis Tutorial  Teaching Game Tutorial

Installation

  • See the releases page for downloadable executables for Windows and macOS.
  • Alternatively use pip3 install -U katrain to install the latest version from PyPI on any 64-bit OS.
  • On macOS, you can also use brew install katrain to install the app.
  • This page has detailed instructions for Window, Linux and macOS, as well as troubleshooting and setting up KataGo to use multiple GPUs.

Configuring KataGo

KaTrain comes pre-packaged with a working KataGo (OpenCL version) for Windows, Linux, and pre-M1 Mac operating systems, and the rather old 15 block model.

To change the model, open 'General and Engine settings' in the application and 'Download models'. You can then select the model you want from the dropdown menu.

To change the katago binary, e.g. to the Eigen/CPU version if you don't have a GPU, click 'Download KataGo versions'. You can then select the KataGo binary from the dropdown menu. There are also CUDA and TensorRT versions available on the KataGo release site. Particularly the latter may offer much better performance on NVIDIA GPUs, but will be harder to set up: see here for more details.

Finally, you can override the entire command used to start the analysis engine, which can be useful for connecting to a remote server. Do keep in mind that KaTrain uses the analysis engine of KataGo, and not the GTP engine.

Play against AI

  • Select the players in the main menu, or under 'New Game'.
  • In a teaching game, KaTrain will analyze your moves and automatically undo those that are sufficiently bad.
  • When playing against AI, note that the "Undo" button will undo both the AI's last move and yours.

Instant feedback

The dots on the move indicate how many points were lost by that move.

  • The colour indicates the size of the mistake according to KataGo
  • The size indicates if the mistake was actually punished. Going from fully punished at maximal size, to no actual effect on the score at minimal size.

In short, if you are a weaker player you should mostly focus on large dots that are red or purple, while stronger players can pay more attention to smaller mistakes. If you want to hide some colours on the board, or not output details for them in SGFs,you can do so under 'Configure Teacher'.

AIs

This section describes the available AIs.

In the 'AI settings', settings which have been tested and calibrated are at the top and have a lighter color, changing these will show an estimate of rank. This estimate should be reasonably accurate as long as you have not changed the other settings.

  • Recommended options for serious play include:
    • KataGo is full KataGo, above professional level. The analysis and feedback given is always based on this full strength KataGo AI.
    • Calibrated Rank Bot was calibrated on various bots (e.g. GnuGo and Pachi at different strength settings) to play a balanced game from the opening to the endgame without making serious (DDK) blunders. Further discussion can be found here and here.
    • Simple Style Prefers moves that solidify both player's territory, leading to relatively simpler moves.
  • Legacy options which were developed earlier include:
    • ScoreLoss is KataGo analyzing as usual, but choosing from potential moves depending on the expected score loss, leading to a varied style with mostly small mistakes.
    • Policy uses the top move from the policy network (it's 'shape sense' without reading).
    • Policy Weighted picks a random move weighted by the policy, leading to a varied style with mostly small mistakes, and occasional blunders due to a lack of reading.
    • Blinded Policy picks a number of moves at random and play the best move among them, being effectively 'blind' to part of the board each turn. Calibrated rank is based on the same idea, and recommended over this option.
  • Options that are more on the 'fun and experimental' side include:
    • Variants of Blinded Policy, which use the same basic strategy, but with a twist:
      • Local Style will consider mostly moves close to the last move.
      • Tenuki Style will consider mostly moves away from the last move.
      • Influential Style will consider mostly 4th+ line moves, leading to a center-oriented style.
      • Territory Style is biased in the opposite way, towards 1-3rd line moves.
    • KataJigo is KataGo attempting to win by 0.5 points, typically by responding to your mistakes with an immediate mistake of it's own.
    • KataAntiMirror is KataGo assuming you are playing mirror go and attempting to break out of it with profit as long as you are.

The Engine based AIs (KataGo, ScoreLoss, KataJigo) are affected by both the model and choice of visits and maximum time, while the policy net based AIs are affected by the choice of model file, but work identically with 1 visit.

Further technical details and discussion on some of these AIs can be found on this thread at the life in 19x19 forums.

Analysis

Analysis options in KaTrain allow you to explore variations and request more in-depth analysis from the engine at any point in the game.

Key Short Description Details
Tab Switch between analysis and play modes AI moves, teaching mode and timers are suspended in analysis mode. The state of the analysis options and right-hand side panels and options is saved independently for 'play' and 'analyze', allowing you to quickly switch between a more minimalistic 'play' mode and more complex 'analysis' mode.

The checkboxes at the top of the screen:

Key Short Description Details
q Child moves are shown On by default, can turn it off to avoid obscuring other information or when wanting to guess the next move.
w Show all dots Toggles showing coloured evaluation 'dots' on the last few moves or not. You can configure the thresholds, along with how many of the last moves they are shown for under 'Teaching/Analysis Settings'.
e Top moves Show the next moves KataGo considered, colored by their expected point loss. Small/faint dots indicate high uncertainty and never show text (lower than your 'fast visits' setting). Hover over any of them to see the principal variation.
r Policy moves Show KataGo's policy network evaluation, i.e. where it thinks the best next move is purely from the position, and in the absence of any 'reading'. This turns off the 'top moves' setting as the overlap is often not useful.
t Expected territory Show expected ownership of each intersection.

The analysis options available under the 'Analysis' button are used for deeper evaluation of the position:

Key Short Description Details
a Deeper analysis Re-evaluate the position using more visits, usually resulting in a more accurate evaluation.
s Equalize visits Re-evaluate all currently shown next moves with the same visits as the current top move. Useful to increase confidence in the suggestions with high uncertainty.
d Analyze all moves Evaluate all possible next moves. This can take a bit of time even though 'fast_visits' is used, but can be useful to see how many reasonable next moves are available.
f Find alternatives Increases analysis of current candidate moves to at least the 'fast visits' level, and request a new query that excludes all current candidate moves.
g Select area of interest Set an area and search only for moves in this box. Good for solving tsumegos. Note that some results may appear outside the box due to establishing a baseline for the best move, and the opponent can tenuki in variations.
h Reset analysis This reverts the analysis to what the engine returns after a normal query, removing any additional exploration.
i Start insertion mode Allows you to insert moves, to improve analysis when both players ignore an important exchange or life and death situation. Press again to stop inserting and copy the rest of the branch.
l Play out the game until the end and add as a collapsed branch, to visualize the potential effect of mistakes This is done in the background, and can be started at several nodes at once when comparing the results at different starting positions.
Space Turn continuous analysis on/off. This will continuously improve analysis of the current position, similar to Lizzie's 'pondering', but only when there are no other queries going on.
Shift + Space As above, but does not turn 'top moves' hints on when it is off.
Enter AI move Makes the AI move for the current player regardless of current player selection.
F2 Deeper full game analysis Analyze the entire game to a higher number of visits.
F3 Performance report Show an overview of performance statistics for both players.
F10 Tsumego Frame After placing a life and death problem in a corner/side, use this to fill up the rest of the board to improve AI's ability in solving life and death problems.

Keyboard and mouse shortcuts

In addition to shortcuts mentioned above and those shown in the main menu:

Key Short Description Details
Alt Open the main menu
~ or ` or F12 Cycles through more minimalistic UI modes
k Toggle display of board coordinates
p Pass
Pause Pause/Resume timer
or z Undo move Hold shift for 10 moves at a time, or ctrl to skip to the start.
or x Redo move Hold shift for 10 moves at a time, or ctrl to skip to the end.
/ Switch branch As would be expected from the move tree.
Home/End Go to the beginning/end of the game
PageUp Make the currently selected node the main branch
Ctrl + Delete Delete current node
c Collapse/Uncollapse the branch from the current node to the previous branching point
b Go back to the previous branching point
Shift + b Go back the main branch
n Go to one move before the next mistake (orange or worse) by a human player As in clicking the forward red arrow
Shift + n Go to one move before the previous mistake As in clicking the backward red arrow
Scroll Mouse Redo/Undo move or Scroll through principal variation When hovering the cursor over the right panel: Redo/Undo move. When hovering over a candidate move: Scroll through principal variation.
Middle Scroll Wheel Click Add principal variation to the move tree When scrolling, only moves up to the point you are viewing are added.
Click on a Move See detailed statistics for a previous move Along with expected variation that was best instead of this move
Double Click on a Move Navigate directly to just before that point in the game
Ctrl + v Load SGF from the clipboard and do a 'fast' analysis of the game With a high priority normal analysis for the last move.
Ctrl + c Save SGF to clipboard
Escape Stop all analysis

Contributing to distributed training

Starting in December 2020, KataGo started distributed training. This allows people to all help generate self-play games to increase KataGo's strength and train bigger models.

KaTrain 1.8.0+ makes it easy to contribute to distributed training: simply select the option from the main menu, register an account, and click run. During this mode you can do little more than watch games.

Keep in mind that partial games are not uploaded, so it is best to plan to keep it running for at least an hour, if not several, for the most effective contribution.

A few keyboard shortcuts have special functions in this mode:

Key Short Description Details
Space Switch between manually navigating the current game And automatically advancing it.
Escape Sends the quit command to KataGo Which starts a slow shutdown, finishing partial games but not starting new ones. Only works on v1.11+.
Pause Pauses/resumes contributions via the pause and resume commands Introduced in KataGo v1.11

Themes

See these instructions for how to modify the look of any graphics or colours, and creating or install themes.

FAQ

  • The program is running too slowly. How can I speed it up?
    • Adjust the number of visits or maximum time allowed in the settings.
  • KataGo crashes with "out of memory" errors, how can I prevent this?
    • Try using a lower number for nnMaxBatchSize in KataGo/analysis_config.cfg, and avoid using versions compiled with large board sizes.
    • If still encountering problems, please start KataGo by itself to check for any errors it gives.
    • Note that if you don't have a GPU, or your GPU does not support OpenCL, you should use the 'eigen' binaries which run on CPU only.
  • The font size is too small
    • On some ultra-high resolution monitors, dialogs and other elements with text can appear too small. Please see these instructions to adjust them.
  • The app crashes with an error about "unable to find any valuable cutbuffer provider"
    • Install xclip using sudo apt-get install xclip

Support / Contribute

GitHub issues Contributors

  • Ideas, feedback, and contributions to code or translations are all very welcome.
    • For suggestions and planned improvements, see open issues on github to check if the functionality is already planned.
  • You can join the Computer Go Community Discord (formerly Leela Zero & Friends) (use the #gui channel) to get help, discuss improvements, or simply show your appreciation. Please do not use github issues to ask for technical help, this is only for bugs, suggestions and discussing contributions.

For Tasks:

Click tags to check more tools for each tasks

For Jobs:

Alternative AI tools for katrain

Similar Open Source Tools

For similar tasks

For similar jobs