| import os | |
| params_height = 256 | |
| params_width = 256 | |
| params_m = 32 | |
| params_number_input = 1 | |
| params_step_size = 5 | |
| params_gamma = 0.2 | |
| TRAIN_LOCATION = "./lf_train.txt" | |
| VALIDATION_LOCATION = "./lf_validate.txt" | |
| TEST_LOCATION = "./lf_test.txt" | |
| LOG_FILE_LOCATION = "./logs/training_log_0.txt" | |
| CHECKPOINT_LOCATION = "./checkpoint/" | |
| RESUME_CHECKPOINT_LOCATION = "./checkpoint/checkpoint_best.pth" | |
| START_CHECKPOINT_LOCATION = "./checkpoint/checkpoint_init.pth" | |
| DEVICE = "cpu" | |
| BATCH_SIZE = 24 | |
| LEARNING_RATE = 0.00001 | |
| NUM_EPOCHS = 50 | |
| START_EPOCH = 0 | |
| T_max = 50 | |
| PRINT_INTERVAL = 20 | |
| os.makedirs("./logs",exist_ok=True) | |
| os.makedirs("./checkpoint",exist_ok=True) | |
| os.makedirs("./output",exist_ok=True) | |