| import os | |
| params_height = 256 | |
| params_width = 256 | |
| params_m = 32 | |
| params_number_input = 1 | |
| params_step_size = 2 | |
| params_gamma = 0.2 | |
| DEVICE = "cpu" | |
| os.makedirs("./logs",exist_ok=True) | |
| os.makedirs("./checkpoint",exist_ok=True) | |
| os.makedirs("./output",exist_ok=True) | |