| Geant4 Cross Reference |
1 """ 1 """
2 ** setup ** 2 ** setup **
3 creates necessary folders 3 creates necessary folders
4 """ 4 """
5 5
6 import os 6 import os
7 7
8 from core.constants import INIT_DIR, GLOBAL_CH 8 from core.constants import INIT_DIR, GLOBAL_CHECKPOINT_DIR, CONV_DIR, VALID_DIR, GEN_DIR
9 9
10 for folder in [INIT_DIR, # Directory to load 10 for folder in [INIT_DIR, # Directory to load the full simulation dataset
11 GLOBAL_CHECKPOINT_DIR, # Direc 11 GLOBAL_CHECKPOINT_DIR, # Directory to save VAE checkpoints
12 CONV_DIR, # Directory to save 12 CONV_DIR, # Directory to save model after conversion to a format that can be used in C++
13 VALID_DIR, # Directory to save 13 VALID_DIR, # Directory to save validation plots
14 GEN_DIR, # Directory to save V 14 GEN_DIR, # Directory to save VAE generated showers
15 ]: 15 ]:
16 os.system(f"mkdir {folder}") 16 os.system(f"mkdir {folder}")