SUGA-ya

you can use parameter "generate_test_images" in main.py, change the value whatever you like change the visualize function in utils.py like this:(about line 234) elif option == 1: values = np.arange(0,...