Update models/test_model.py

models/test_model.py

@@ -5,20 +5,16 @@ from . import networks
 class TestModel(BaseModel):
     """ This TesteModel can be used to generate CycleGAN results for only one direction.
     This model will automatically set '--dataset_mode single', which only loads the images from one collection.
-
     See the test instruction for more details.
     """
     @staticmethod
     def modify_commandline_options(parser, is_train=True):
         """Add new dataset-specific options, and rewrite default values for existing options.
-
         Parameters:
             parser          -- original option parser
             is_train (bool) -- whether training phase or test phase. You can use this flag to add training-specific or test-specific options.
-
         Returns:
             the modified parser.
-
         The model can only be used during test time. It requires '--dataset_mode single'.
         You need to specify the network using the option '--model_suffix'.
         """
@@ -30,7 +26,6 @@ class TestModel(BaseModel):
 
     def __init__(self, opt):
         """Initialize the pix2pix class.
-
         Parameters:
             opt (Option class)-- stores all the experiment flags; needs to be a subclass of BaseOptions
         """
@@ -51,10 +46,8 @@ class TestModel(BaseModel):
 
     def set_input(self, input):
         """Unpack input data from the dataloader and perform necessary pre-processing steps.
-
         Parameters:
             input: a dictionary that contains the data itself and its metadata information.
-
         We need to use 'single_dataset' dataset mode. It only load images from one domain.
         """
         self.real = input['A'].to(self.device)