| import os | |
| params_width = 512 | |
| params_height = 352 | |
| 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 = 16 | |
| LEARNING_RATE = 0.0001 | |
| NUM_EPOCHS = 150 | |
| START_EPOCH = 0 | |
| PRINT_INTERVAL = 20 | |
| os.makedirs("./logs",exist_ok=True) | |
| os.makedirs("./checkpoint",exist_ok=True) | |
| os.makedirs("./output",exist_ok=True) | |