--- a +++ b/params/test_params.py @@ -0,0 +1,21 @@ +from .basic_params import BasicParams + + +class TestParams(BasicParams): + """ + This class is a son class of BasicParams. + This class includes parameters for testing and parameters inherited from the father class. + """ + def initialize(self, parser): + parser = BasicParams.initialize(self, parser) + + # Testing parameters + parser.add_argument('--save_latent_space', action='store_true', help='save the latent space of input data to disc') + + # Logging and visualization + parser.add_argument('--print_freq', type=int, default=1, + help='frequency of showing results on console') + + self.isTrain = False + self.isTest = True + return parser