Orion version 0.3.2 does not support TadGAN to change `batch_size`. #### What needs to be changed https://github.com/sintel-dev/Orion/blob/192f45af8a5eae49ad57503b10570f12103574bf/orion/primitives/tadgan.py#L31 Allow a variable input ```python3 alpha = K.random_uniform((batch_size, 1, 1)) ```