Currently, for my convenience, I have fixed some parameters for some gpu platform (e.g., I have to mount different storage) The dataset path in main.py The output path in main.py The model weights.