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)