Add auto batch size calculation in training scripts

[hvgazula]

• code snippet in test_train.py
• implement it in brainy_train.py and kwyk_train.py as a decorator (maybe)
• this is very useful for kwyk training which is trained on small blocks and can be inefficient with batch_size = 1