baby.training.utils

Functions

augmented_generator(gen, aug)
fix_tf_rtx_gpu_bug()	Run to set up TensorFlow session with RTX series NVidia GPUs

Classes

BabyTrainerParameters([...])	Global parameters for training BABY models
TrainValProperty(train, val)	Attributes
TrainValTestProperty(train, val, test)	Attributes