Add data augmentation: hflip, vflip, rotation (90, 180, 270) Add marker for ko/unplayable move Code a better generator/faster generator (rn generator is the bottleneck)