Neuroimaging-Example/src/const_model.py

import os

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
base_path = os.getcwd() + "/data/raw/*/*"

batch_size=4
epochs=1

model_weights_path = "/model_weights/model_weights.h5"