lisiping0817

 In srgan code ,Finally, a common MSE loss is used?

image_stylization_create_dataset.py

3

this file maybe lack of console_entry_point method ,so when we from magenta.models.image_stylization.image_stylization_create_dataset import console_entry_point ,there will a importerror.

training.py should import main.py function .such as MSE = loss_func(sr_result, imgHR)