training/core/constants.py

0001 from utils.optimizer import OptimizerType
0002
0003 """
0004 Experiment constants.
0005 """
0006 # Number of calorimeter layers (z-axis segmentation).
0007 N_CELLS_Z = 45
0008 # Segmentation in the r,phi direction.
0009 N_CELLS_R = 18
0010 N_CELLS_PHI = 50
0011 # Cell size in the r and z directions
0012 SIZE_R = 2.325
0013 SIZE_Z = 3.4
0014
0015 # Minimum and maximum primary particle energy to consider for training in GeV units.
0016 MIN_ENERGY = 1
0017 MAX_ENERGY = 1024
0018 # Minimum and maximum primary particle angle to consider for training in degrees units.
0019 MIN_ANGLE = 50
0020 MAX_ANGLE = 90
0021
0022 """
0023 Directories.
0024 """
0025 # Directory to load the full simulation dataset.
0026 INIT_DIR = "./dataset/"
0027 # Directory to save VAE checkpoints
0028 GLOBAL_CHECKPOINT_DIR = "./checkpoint"
0029 # Directory to save model after conversion to a format that can be used in C++.
0030 CONV_DIR = "./conversion"
0031 # Directory to save validation plots.
0032 VALID_DIR = "./validation"
0033 # Directory to save VAE generated showers.
0034 GEN_DIR = "./generation"
0035
0036 """
0037 Model default parameters.
0038 """
0039 BATCH_SIZE_PER_REPLICA = 128
0040 # Total number of readout cells (represents the number of nodes in the input/output layers of the model).
0041 ORIGINAL_DIM = N_CELLS_Z * N_CELLS_R * N_CELLS_PHI
0042 INTERMEDIATE_DIMS = [100, 50, 20, 14]
0043 LATENT_DIM = 10
0044 EPOCHS = 1000
0045 LEARNING_RATE = 0.001
0046 ACTIVATION = "leaky_relu"
0047 OUT_ACTIVATION = "sigmoid"
0048 VALIDATION_SPLIT = 0.10
0049 NUMBER_OF_K_FOLD_SPLITS = 1
0050 OPTIMIZER_TYPE = OptimizerType.ADAM
0051 KERNEL_INITIALIZER = "RandomNormal"
0052 BIAS_INITIALIZER = "Zeros"
0053 EARLY_STOP = False
0054 SAVE_BEST_MODEL = True
0055 SAVE_MODEL_EVERY_EPOCH = True
0056 PATIENCE = 10
0057 MIN_DELTA = 0.01
0058 BEST_MODEL_FILENAME = "VAE_best"
0059 # GPU identifiers separated by comma, no spaces.
0060 GPU_IDS = "0"
0061 # Maximum allowed memory on one of the GPUs (in GB)
0062 MAX_GPU_MEMORY_ALLOCATION = 32
0063 # Buffer size used while shuffling the dataset.
0064 BUFFER_SIZE = 1000
0065
0066 """
0067 Optimizer parameters.
0068 """
0069 N_TRIALS = 50
0070 # Maximum size of a hidden layer
0071 MAX_HIDDEN_LAYER_DIM = 2000
0072
0073 """
0074 Validator parameter.
0075 """
0076 FULL_SIM_HISTOGRAM_COLOR = "blue"
0077 ML_SIM_HISTOGRAM_COLOR = "red"
0078 FULL_SIM_GAUSSIAN_COLOR = "green"
0079 ML_SIM_GAUSSIAN_COLOR = "orange"
0080 HISTOGRAM_TYPE = "step"
0081
0082 """
0083 W&B parameters.
0084 """
0085 # Change this to your entity name.
0086 WANDB_ENTITY = "entity-name"