Neuroimaging-Example/src/const_eval.py

input_shape: tuple = (4, 80, 96, 64)
# The paper is using (4, 160, 192, 128) with a batch size of 1 to fit the GPU memory.
# We will use a smaller image with a larger batch size because of time & memory issues.
output_channels: int = 3