Spaces:
Build error
Build error
| import tensorflow as tf | |
| # DATA | |
| DATASET_NAME = "organmnist3d" | |
| BATCH_SIZE = 32 | |
| AUTO = tf.data.AUTOTUNE | |
| INPUT_SHAPE = (28, 28, 28, 1) | |
| NUM_CLASSES = 11 | |
| # OPTIMIZER | |
| LEARNING_RATE = 1e-4 | |
| WEIGHT_DECAY = 1e-5 | |
| # TRAINING | |
| EPOCHS = 80 | |
| # TUBELET EMBEDDING | |
| PATCH_SIZE = (8, 8, 8) | |
| NUM_PATCHES = (INPUT_SHAPE[0] // PATCH_SIZE[0]) ** 2 | |
| # ViViT ARCHITECTURE | |
| LAYER_NORM_EPS = 1e-6 | |
| PROJECTION_DIM = 128 | |
| NUM_HEADS = 8 | |
| NUM_LAYERS = 8 | |
| labels = ['liver', 'kidney-right', 'kidney-left', 'femur-right', 'femur-left', 'bladder', 'heart', 'lung-right', 'lung-left', 'spleen', 'pancreas'] | |