In the code : train_model.py
""" test for generator """
loss_d = self.forward_discriminator(batch_samples, RNA_input_dim, ATAC_input_dim, d_loss, 'test')
reconstruct_loss, kl_div, loss_g = self.forward_translator(batch_samples, RNA_input_dim, ATAC_input_dim, a_loss, r_loss, weight_temp, 'train', kl_mean)
I noticed that the parameter is set to "train" in the forward_translator function, but the context seems to indicate that it is in the "test" phase for the discriminator (forward_discriminator). Would it be possible to clarify this?
In the code : train_model.py
""" test for generator """
loss_d = self.forward_discriminator(batch_samples, RNA_input_dim, ATAC_input_dim, d_loss, 'test')
reconstruct_loss, kl_div, loss_g = self.forward_translator(batch_samples, RNA_input_dim, ATAC_input_dim, a_loss, r_loss, weight_temp, 'train', kl_mean)
I noticed that the parameter is set to "train" in the forward_translator function, but the context seems to indicate that it is in the "test" phase for the discriminator (forward_discriminator). Would it be possible to clarify this?