968ed8b75cc183e66777ca9d75ea765eede7a57c,autokeras/nn/model_trainer.py,GANModelTrainer,__init__,#GANModelTrainer#Any#Any#Any#Any#Any#Any#,230

Before Change

        self.out_f = None
        self.out_size = 0
        if gen_training_result is not None:
            self.out_f, self.out_size = gen_training_result
            self.sample_noise = torch.randn(self.out_size,
                                            self.g_model.nz,
                                            1, 1, device=self.device)

After Change

                 metric=None,
                 verbose=False,
                 device=None):
        if device is not None:
            self.device = device
        else:
            self.device = get_device()
        self.metric = metric
        self.verbose = verbose
        self.loss_function = loss_function
        self.train_loader = train_data

Instances